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 6c9a608 commit e6e9e9bCopy full SHA for e6e9e9b
efficientnetv2/infer.py
@@ -48,8 +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 = copy.deepcopy(hparams.base_config)
52
- config.override(effnetv2_configs.get_model_config(model_name))
+ config = effnetv2_configs.get_model_config(model_name)
53
config.override(datasets.get_dataset_config(dataset_cfg))
54
config.override(hparam_str)
55
config.model.num_classes = config.data.num_classes
0 commit comments