kritsg commited on
Commit
4be4595
·
1 Parent(s): 355d154

removed print statements

Browse files
Files changed (1) hide show
  1. bayes/models.py +0 -1
bayes/models.py CHANGED
@@ -77,7 +77,6 @@ def process_imagenet_get_model(data):
77
  for q in range(0, perturbed_images.shape[0], batch_size):
78
  predictions.append(softmax(model(perturbed_images[q:q+batch_size])).cpu().detach().numpy())
79
  predictions = np.concatenate(predictions, axis=0)
80
- print("Predictions made in models.py:", predictions)
81
  return predictions
82
  return wrapped_model
83
 
 
77
  for q in range(0, perturbed_images.shape[0], batch_size):
78
  predictions.append(softmax(model(perturbed_images[q:q+batch_size])).cpu().detach().numpy())
79
  predictions = np.concatenate(predictions, axis=0)
 
80
  return predictions
81
  return wrapped_model
82