From 7049459959f46295e9afd1bcd05d0fa147e703ac Mon Sep 17 00:00:00 2001
From: Huaiyu Miao <miaohuaiyu@ruc.edu.cn>
Date: Sun, 5 Apr 2020 14:08:40 +0800
Subject: [PATCH] Update train.py

add fcn_model.train() to line 87
---
 python/train.py | 1 +
 1 file changed, 1 insertion(+)

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()