Skip to content
Snippets Groups Projects
Commit 5e156024 authored by Alexandre Strube's avatar Alexandre Strube
Browse files

6 epochs is enough

parent f8c22e71
Branches
No related tags found
No related merge requests found
Pipeline #140683 passed
...@@ -13,4 +13,4 @@ dls = DataBlock( ...@@ -13,4 +13,4 @@ dls = DataBlock(
learn = Learner(dls, xresnet50(n_out=10), metrics=[accuracy,top_k_accuracy]).to_fp16() learn = Learner(dls, xresnet50(n_out=10), metrics=[accuracy,top_k_accuracy]).to_fp16()
with learn.distrib_ctx(): with learn.distrib_ctx():
learn.fine_tune(12) learn.fine_tune(6)
...@@ -13,4 +13,4 @@ dls = DataBlock( ...@@ -13,4 +13,4 @@ dls = DataBlock(
learn = Learner(dls, xresnet50(n_out=10), metrics=[accuracy,top_k_accuracy]).to_fp16() learn = Learner(dls, xresnet50(n_out=10), metrics=[accuracy,top_k_accuracy]).to_fp16()
learn.fine_tune(48) learn.fine_tune(6)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment