diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 0906d58..a5d898a 100644 --- 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