Update model.py

run prepush.sh, sorry!
czbiohub-sf · Jun 13, 2024 · a5cda91 · a5cda91
1 parent 26307dd
commit a5cda91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yogo/model.py b/yogo/model.py
@@ -143,7 +143,7 @@ def from_pth(
         return model, {
             "step": global_step,
             "class_names": class_names,
-            "normalize_images": normalize_images
+            "normalize_images": params["normalize_images"],
         }
 
     def to(self, device, *args, **kwargs):