Fix missing cfg key check

pull/1/head
Ross Wightman 5 years ago
parent 073d31a076
commit c88e80081d

@ -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

Loading…
Cancel
Save