Fix missing cfg key check

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

@ -61,7 +61,7 @@ def resolve_data_config(model, args, default_cfg={}, verbose=True):
else: else:
assert len(std) == in_chans assert len(std) == in_chans
new_config['std'] = std new_config['std'] = std
else: elif 'std' in default_cfg:
new_config['std'] = default_cfg['std'] new_config['std'] = default_cfg['std']
# resolve default crop percentage # resolve default crop percentage

Loading…
Cancel
Save