diff options
| -rw-r--r-- | util/parse_config.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/util/parse_config.py b/util/parse_config.py index 00a4fd4..39b9c13 100644 --- a/util/parse_config.py +++ b/util/parse_config.py @@ -15,6 +15,8 @@ def parse_model_config(yaml): config['use_positional'] = yaml['use_positional'] if 'use_feedforward' in yaml: config['use_feedforward'] = yaml['use_feedforward'] + if 'use_attention' in yaml: + config['use_attention'] = yaml['use_attention'] return config |