Fix saving

radionets-project · Feb 20, 2023 · 3aaa208 · 3aaa208
1 parent edd9900
commit 3aaa208
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/radionets/dl_framework/model.py b/radionets/dl_framework/model.py
@@ -171,7 +171,7 @@ def load_pre_model(learn, pre_path, visualize=False, plot_loss=False):
 
 
 def save_model(learn, model_path):
-    if hasattr(learn, "normalize"):
+    if hasattr(learn, "mean_real"):
         norm_dict = {
             "mean_real": learn.normalize.mean_real,
             "mean_imag": learn.normalize.mean_imag,