From 5e1560243c8c8a7ee99cab676e5246a38e5988b1 Mon Sep 17 00:00:00 2001
From: Alexandre Strube <a.strube@fz-juelich.de>
Date: Mon, 29 May 2023 20:34:28 +0200
Subject: [PATCH] 6 epochs is enough

---
 src/distrib.py | 2 +-
 src/serial.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/distrib.py b/src/distrib.py
index c3b45b8..322a146 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 a967563..ed1d29d 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)
-- 
GitLab