fixed classification model name typo

uwmisl · May 20, 2022 · b00d8d7 · b00d8d7
1 parent 56a32c7
commit b00d8d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/classification/all_36_barcodes_cnn_training.ipynb b/classification/all_36_barcodes_cnn_training.ipynb
@@ -500,7 +500,7 @@
     "'''\n",
     "Saving the trained net\n",
     "'''\n",
-    "torch.save(net.state_dict(), \"../utils/model/all_36_barcodes_cnn_results_20220519.pt\")"
+    "torch.save(net.state_dict(), \"../utils/model/all_36_barcodes_trained_cnn_20220519.pt\")"
    ]
   }
  ],

diff --git a/classification/all_36_barcodes_cnn_training.py b/classification/all_36_barcodes_cnn_training.py
@@ -235,5 +235,5 @@
 '''
 Saving the trained net
 '''
-torch.save(net.state_dict(), "../utils/model/all_36_barcodes_cnn_results_20220519.pt")
+torch.save(net.state_dict(), "../utils/model/all_36_barcodes_trained_cnn_20220519.pt")