diff options
| author | Marcin Chrzanowski <mc370754@students.mimuw.edu.pl> | 2021-05-29 19:36:21 +0200 |
|---|---|---|
| committer | Marcin Chrzanowski <mc370754@students.mimuw.edu.pl> | 2021-05-29 19:36:21 +0200 |
| commit | 403c235c0b1dee7fbb314c6d2d13e484ec7cc889 (patch) | |
| tree | e2e7e75b0486d38751a60b93ad969d4cb42bd33c /util | |
| parent | 249de72e67db2027c8207e84c88785b3f0782362 (diff) | |
Allow attention configuration
Diffstat (limited to 'util')
| -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 |