From 5787951e1d62c12fb7fd6ec7f2ebcda484858680 Mon Sep 17 00:00:00 2001 From: Gerome Meyer Date: Wed, 15 May 2024 14:48:48 +0200 Subject: [PATCH] Training for longer duration --- training/src/configurations/TrainConfig.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training/src/configurations/TrainConfig.py b/training/src/configurations/TrainConfig.py index 1ca5a4d..b4e01a0 100644 --- a/training/src/configurations/TrainConfig.py +++ b/training/src/configurations/TrainConfig.py @@ -7,5 +7,5 @@ class TrainConfig(Configuration): train_batch_size = 96 eval_batch_size = 8 - num_epochs = 30 + num_epochs = 60 num_images = 7351