Skip to content

Instantly share code, notes, and snippets.

@neelriyer
Created July 5, 2020 10:05
Show Gist options
  • Save neelriyer/6568b3ccd35fc943e34527d6dbe69740 to your computer and use it in GitHub Desktop.
Save neelriyer/6568b3ccd35fc943e34527d6dbe69740 to your computer and use it in GitHub Desktop.
Train Model on small images (128x128)
lr = 1e-2
def do_fit(save_name, lrs=slice(lr), pct_start=0.9, cycles = 10):
learn.fit_one_cycle(cycles, lrs, pct_start=pct_start)
learn.save(save_name)
learn.show_results(rows=2, imgsize=7)
do_fit('1a', slice(lr))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment