diff --git a/yogo/model.py b/yogo/model.py index bf9efbee..929314f6 100644 --- a/yogo/model.py +++ b/yogo/model.py @@ -143,7 +143,6 @@ def from_pth( return model, { "step": global_step, "class_names": class_names, - "normalize_images": params["normalize_images"], } def to(self, device, *args, **kwargs):