From 341824fb06864ce999103afbffccab73cbc808f6 Mon Sep 17 00:00:00 2001
From: milesial <milesial@users.noreply.github.com>
Date: Wed, 26 Sep 2018 08:58:49 +0200
Subject: [PATCH]  Switch net to train mode in train.py

Former-commit-id: ab255c4b3b823dfe8681083aeac0ee78c1a54780
---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index 3b0e4d6..a470606 100644
--- a/train.py
+++ b/train.py
@@ -54,6 +54,7 @@ def train_net(net,
 
     for epoch in range(epochs):
         print('Starting epoch {}/{}.'.format(epoch + 1, epochs))
+        net.train()
 
         # reset the generators
         train = get_imgs_and_masks(iddataset['train'], dir_img, dir_mask, img_scale)