diff --git a/python/train.py b/python/train.py
index f379bc3..366411f 100644
--- a/python/train.py
+++ b/python/train.py
@@ -84,6 +84,7 @@
 
 def train():
     for epoch in range(epochs):
+        fcn_model.train()
         scheduler.step()
 
         ts = time.time()