diff --git a/src/distrib.py b/src/distrib.py index c3b45b8f28622cd0a08983778b45e7aecf540bb2..322a146ad353a0c3d827b4498a765d1db7eff264 100644 --- a/src/distrib.py +++ b/src/distrib.py @@ -13,4 +13,4 @@ dls = DataBlock( learn = Learner(dls, xresnet50(n_out=10), metrics=[accuracy,top_k_accuracy]).to_fp16() with learn.distrib_ctx(): - learn.fine_tune(12) + learn.fine_tune(6) diff --git a/src/serial.py b/src/serial.py index a967563ca3ce373814da0e85a48e966bf9856a06..ed1d29dacb54da42f59b5127e111c0b256e2e501 100644 --- a/src/serial.py +++ b/src/serial.py @@ -13,4 +13,4 @@ dls = DataBlock( learn = Learner(dls, xresnet50(n_out=10), metrics=[accuracy,top_k_accuracy]).to_fp16() -learn.fine_tune(48) +learn.fine_tune(6)