diff --git a/python/train.py b/python/train.py index 036eb69f5c346e720388e4635ef6e3975cb68498..a9c94ec6807e14e3c9a39ceb816f619fc1b37268 100644 --- a/python/train.py +++ b/python/train.py @@ -12,6 +12,7 @@ import os #- Update the weights for different convolution layers based on the gradients. def validate(model, valloader, use_cuda, epoch): + model.eval() correct = 0 total = 0 for idx, (inputs, targets) in enumerate(valloader):