lesscomfortable
commited on
Commit
•
26c3b11
1
Parent(s):
2077d78
Update torch_utils.py (#751)
Browse files- 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
|