Skip to content

Commit 42ad3a4

Browse files
committed
Simple fix.
1 parent e6e9e9b commit 42ad3a4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

efficientnetv2/infer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def define_flags():
4848

4949
def get_config(model_name, dataset_cfg, hparam_str=''):
5050
"""Create a keras model for EffNetV2."""
51-
config = effnetv2_configs.get_model_config(model_name)
51+
config = copy.deepcopy(effnetv2_configs.get_model_config(model_name))
5252
config.override(datasets.get_dataset_config(dataset_cfg))
5353
config.override(hparam_str)
5454
config.model.num_classes = config.data.num_classes

0 commit comments

Comments
 (0)