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