Switch net to train mode in train.py

Former-commit-id: ab255c4b3b823dfe8681083aeac0ee78c1a54780
2018-09-26 08:58:49 +02:00 · 2018-09-26 08:58:49 +02:00 · 341824fb06
parent b1ac7151b4
commit 341824fb06
1 changed files with 1 additions and 0 deletions
--- 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)