From 74313864276eac77bb1802c8cb5ede9f8be2721f Mon Sep 17 00:00:00 2001
From: swarajpande4 <swarajpande4@gmail.com>
Date: Wed, 24 May 2023 11:43:07 +0530
Subject: [PATCH] Fixed epochs and sample_interval

---
 code/train_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/code/train_model.py b/code/train_model.py
index f0e41cc..d109707 100644
--- a/code/train_model.py
+++ b/code/train_model.py
@@ -13,7 +13,7 @@ def train_model():
     
     start_time = time.time()
     acgan = ACGAN() 
-    acgan.train(epochs=120, batch_size=32, sample_interval=10)
+    acgan.train(epochs=125, batch_size=32, sample_interval=25)
     end_time = time.time()
     
     with open('./metrics/train_metric.json', 'w') as f: