From f3ec207a994f6e64c04185bd1ee7e0c7bbd4dd23 Mon Sep 17 00:00:00 2001 From: MochizukiShinichi Date: Fri, 21 Jun 2019 14:08:46 -0700 Subject: [PATCH] Fix inconsistent documentations --- block_zoo/Transformer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/block_zoo/Transformer.py b/block_zoo/Transformer.py index f96f794..f3868d5 100644 --- a/block_zoo/Transformer.py +++ b/block_zoo/Transformer.py @@ -17,11 +17,11 @@ class TransformerConf(BaseConf): Args: attention (str): attention name attention_conf (dict): configurations of attention - layernorm1 (str): layernorm1 name + layernorm_1 (str): layernorm1 name layernorm1_conf (dict): configurations of layernorm1 mlp (str): mlp name mlp_conf (dict): configuration of mlp - layernorm2 (str): layernorm2 name + layernorm_2 (str): layernorm2 name layernorm2_conf (dict): configurations of layernorm2 n_layer (int) layer num of transformer