Update torch_utils.py (#751)

5 years ago · 26c3b11fdc
parent 2077d78880
commit 26c3b11fdc
1 changed files with 1 additions and 1 deletions
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@ -159,7 +159,7 @@ def load_classifier(name='resnet101', n=2):
    input_range = [0, 1]
    mean = [0.485, 0.456, 0.406]
    std = [0.229, 0.224, 0.225]
-    for x in [input_size, input_space, input_range, mean, std]:
+    for x in ['input_size', 'input_space', 'input_range', 'mean', 'std']:
        print(x + ' =', eval(x))

    # Reshape output to n classes