changing model

bayes/models.py

@@ -42,7 +42,7 @@ def process_imagenet_get_model(data):
     """Gets wrapped imagenet model."""
     # Get the vgg16 model, used in the experiments
     model = EfficientNetB0(weights='imagenet')
-    model.eval()
+    # model.eval()
     # model.cuda()
 
     xtest = data['X']