From 0c6f95a6697916dd7a81365334dd739c912afd99 Mon Sep 17 00:00:00 2001 From: Filippo Date: Tue, 6 Feb 2024 11:22:24 +0100 Subject: [PATCH] epochs --- config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.py b/config.py index 1c0a57a..af6a4c8 100644 --- a/config.py +++ b/config.py @@ -92,7 +92,7 @@ # train parameters resize = 224 lr = 1e-3 -epochs = 50 +epochs = 5 sim_batch = 128 gen_batch = 128