lesscomfortable commited on
Commit
26c3b11
1 Parent(s): 2077d78

Update torch_utils.py (#751)

Browse files
Files changed (1) hide show
  1. utils/torch_utils.py +1 -1
utils/torch_utils.py CHANGED
@@ -159,7 +159,7 @@ def load_classifier(name='resnet101', n=2):
159
  input_range = [0, 1]
160
  mean = [0.485, 0.456, 0.406]
161
  std = [0.229, 0.224, 0.225]
162
- for x in [input_size, input_space, input_range, mean, std]:
163
  print(x + ' =', eval(x))
164
 
165
  # Reshape output to n classes
 
159
  input_range = [0, 1]
160
  mean = [0.485, 0.456, 0.406]
161
  std = [0.229, 0.224, 0.225]
162
+ for x in ['input_size', 'input_space', 'input_range', 'mean', 'std']:
163
  print(x + ' =', eval(x))
164
 
165
  # Reshape output to n classes