Removed commented out code

clamytoe · Dec 18, 2022 · 698794a · 698794a
1 parent 26da4e0
commit 698794a
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/train.py b/train.py
@@ -76,9 +76,6 @@
 lrs = learn.lr_find(suggest_funcs=(minimum, steep, valley, slide))
 print(f"{lrs=}")
 
-# print("Performing initial training...")
-# learn.fit_one_cycle(3, lrs.valley)
-
 save_best = SaveModelCallback(
     monitor="valid_loss",
     min_delta=0.000001,
@@ -96,13 +93,6 @@
 
 print("Model validation:", learn.validate())
 
-# print("Unfreezing the last two layers...")
-# learn.unfreeze()
-
-# # Fine tune the model
-# print("Fine tuning the model...")
-# learn.fine_tune(12, lrs.valley)
-
 # Save the model
 print("Saving model as: fastai_model.pkl...")
 learn.export("fastai_model.pkl")