remove lr plot

SydneyBioX · Sep 7, 2023 · dc5be56 · dc5be56
1 parent 308b182
commit dc5be56
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/bidcell/model/train.py b/bidcell/model/train.py
@@ -299,11 +299,11 @@ def train(config: Config):
         lrs.append(cur_lr)
 
     # Plot lr scheduler
-    plt.plot(lrs, ".-", label="LambdaLR")
-    plt.xlabel("epoch")
-    plt.ylabel("LR")
-    plt.tight_layout()
-    plt.savefig(experiment_path + "/LR.png", dpi=300)
+    # plt.plot(lrs, ".-", label="LambdaLR")
+    # plt.xlabel("epoch")
+    # plt.ylabel("LR")
+    # plt.tight_layout()
+    # plt.savefig(experiment_path + "/LR.png", dpi=300)
 
     logging.info("Training finished")