We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e6e9e9b commit 42ad3a4Copy full SHA for 42ad3a4
efficientnetv2/infer.py
@@ -48,7 +48,7 @@ def define_flags():
48
49
def get_config(model_name, dataset_cfg, hparam_str=''):
50
"""Create a keras model for EffNetV2."""
51
- config = effnetv2_configs.get_model_config(model_name)
+ config = copy.deepcopy(effnetv2_configs.get_model_config(model_name))
52
config.override(datasets.get_dataset_config(dataset_cfg))
53
config.override(hparam_str)
54
config.model.num_classes = config.data.num_classes
0 commit comments