Issue #7: indented wandb call to log metrics to the right level

joeranbosma · Dec 8, 2021 · 2e992d9 · 2e992d9
1 parent 1db558a
commit 2e992d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/hamlyn2021/train/train.py b/src/hamlyn2021/train/train.py
@@ -104,8 +104,8 @@ def train_func(path_base, device=None, wandb_un=None, dataset_type="random"):
         if epoch % 100:
             torch.save(net.state_dict(), os.path.join(path_base, "state_dict_model_unet_{}.pt".format(str(time_elapsed))))
 
-    if wandb_un:
-        wandb.log({"train_loss": all_error, "val_loss":all_val_error})
+        if wandb_un:
+            wandb.log({"train_loss": all_error, "val_loss":all_val_error})
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description='HamlynUNet')