From 698794aceb958f0e0324b27ba8a9a7cd27febb0f Mon Sep 17 00:00:00 2001 From: Martin Uribe Date: Sun, 18 Dec 2022 16:47:26 -0600 Subject: [PATCH] Removed commented out code --- train.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/train.py b/train.py index 0c42632..7f25734 100644 --- 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")