Update README.md
Browse files
README.md
CHANGED
|
@@ -36,7 +36,7 @@ from albumentations import Resize, Compose
|
|
| 36 |
from cellseg_models_pytorch.utils import FileHandler
|
| 37 |
from cellseg_models_pytorch.transforms.albu_transforms import MinMaxNormalization
|
| 38 |
|
| 39 |
-
model.set_inference_mode()
|
| 40 |
|
| 41 |
# Resize to multiple of 32 of your own choosing
|
| 42 |
transform = Compose([Resize(1024, 1024), MinMaxNormalization()])
|
|
|
|
| 36 |
from cellseg_models_pytorch.utils import FileHandler
|
| 37 |
from cellseg_models_pytorch.transforms.albu_transforms import MinMaxNormalization
|
| 38 |
|
| 39 |
+
model.set_inference_mode(mixed_precision=False)
|
| 40 |
|
| 41 |
# Resize to multiple of 32 of your own choosing
|
| 42 |
transform = Compose([Resize(1024, 1024), MinMaxNormalization()])
|