From 30fb8fad67af550271e603adf4c37f9a9bf5607b Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Wed, 26 May 2021 07:37:39 +0000 Subject: [PATCH] Update models --- xautodl/models/xcore.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/xautodl/models/xcore.py b/xautodl/models/xcore.py index e7e6b5b..659755f 100644 --- a/xautodl/models/xcore.py +++ b/xautodl/models/xcore.py @@ -1,6 +1,4 @@ ####################################################### -# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 # -####################################################### # Use module in xlayers to construct different models # ####################################################### from typing import List, Text, Dict, Any @@ -41,8 +39,8 @@ def get_model(config: Dict[Text, Any], **kwargs): norm_cls = super_name2norm[kwargs["norm_cls"]] sub_layers, last_dim = [], kwargs["input_dim"] for i, hidden_dim in enumerate(kwargs["hidden_dims"]): - if last_dim > 1: - sub_layers.append(norm_cls(last_dim, elementwise_affine=False)) + if hidden_dim > 1: + sub_layers.append(norm_cls(hidden_dim, elementwise_affine=False)) sub_layers.append(SuperLinear(last_dim, hidden_dim)) sub_layers.append(act_cls()) last_dim = hidden_dim