Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-11-09 23:12:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Switch net to train mode in train.py

Browse source 
Former-commit-id: ab255c4b3b823dfe8681083aeac0ee78c1a54780
This commit is contained in:
milesial 2018-09-26 08:58:49 +02:00 committed by GitHub
parent b1ac7151b4
commit 341824fb06
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
train.py
View file

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