From c88e80081da150b025402a17c3e8ad771a4111de Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 15 Apr 2019 08:45:31 -0700 Subject: [PATCH] Fix missing cfg key check --- data/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/transforms.py b/data/transforms.py index e7dcee2f..bebbd15a 100644 --- a/data/transforms.py +++ b/data/transforms.py @@ -61,7 +61,7 @@ def resolve_data_config(model, args, default_cfg={}, verbose=True): else: assert len(std) == in_chans new_config['std'] = std - else: + elif 'std' in default_cfg: new_config['std'] = default_cfg['std'] # resolve default crop percentage