From 403c235c0b1dee7fbb314c6d2d13e484ec7cc889 Mon Sep 17 00:00:00 2001 From: Marcin Chrzanowski Date: Sat, 29 May 2021 19:36:21 +0200 Subject: Allow attention configuration --- util/parse_config.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'util') 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 -- cgit v1.2.3