Update StarDist_2D_ZeroCostDL4Mic.ipynb

Fixing the learning_rate recording error
HenriquesLab · Aug 2, 2024 · 4827d1d · 4827d1d
1 parent 22236ff
commit 4827d1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Colab_notebooks/StarDist_2D_ZeroCostDL4Mic.ipynb b/Colab_notebooks/StarDist_2D_ZeroCostDL4Mic.ipynb
@@ -1552,7 +1552,7 @@
         "  writer = csv.writer(f)\n",
         "  writer.writerow(['loss','val_loss', 'learning rate'])\n",
         "  for i in range(len(history.history['loss'])):\n",
-        "    writer.writerow([history.history['loss'][i], history.history['val_loss'][i], history.history['lr'][i]])\n",
+        "    writer.writerow([history.history['loss'][i], history.history['val_loss'][i], history.history['learning_rate'][i]])\n",
         "\n",
         "\n",
         "\n",