Skip to content

Instantly share code, notes, and snippets.

@Yuktha-Majella

Yuktha-Majella/freeze_model

Created June 10, 2021 17:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Yuktha-Majella/67a617971649d588c96497db6571bbe3 to your computer and use it in GitHub Desktop.
Save Yuktha-Majella/67a617971649d588c96497db6571bbe3 to your computer and use it in GitHub Desktop.
Download ZIP
Freezing the trained model and exporting it
Raw
freeze_model
t_cnn1.fit_one_cycle(10,max_lr=slice(1e-5, 1e-4))
t_cnn1.freeze()
t_cnn1.export('/content/drive/MyDrive/CV_Vehicle_classification/model/Bmodel_fastai_resnet50.h5')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment