Skip to content

Commit e6e9e9b

Browse files
committed
Do not use base config. Fix google#1018.
1 parent 6c9a608 commit e6e9e9b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

efficientnetv2/infer.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +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 = copy.deepcopy(hparams.base_config)
52-
config.override(effnetv2_configs.get_model_config(model_name))
51+
config = effnetv2_configs.get_model_config(model_name)
5352
config.override(datasets.get_dataset_config(dataset_cfg))
5453
config.override(hparam_str)
5554
config.model.num_classes = config.data.num_classes

0 commit comments

Comments
 (0)