From fb2777177ff33b97f60154c35e783829ad8402d2 Mon Sep 17 00:00:00 2001 From: jingyong hou Date: Mon, 6 Dec 2021 15:14:21 +0800 Subject: [PATCH] format --- kws/model/kws_model.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/kws/model/kws_model.py b/kws/model/kws_model.py index 56cbf76..2f2b2e2 100644 --- a/kws/model/kws_model.py +++ b/kws/model/kws_model.py @@ -126,12 +126,10 @@ def init_model(configs): classifier_type = configs['classifier']['type'] dropout = configs['classifier']['dropout'] - classifier_base = nn.Sequential( - nn.Linear(hidden_dim, 64), - nn.ReLU(), - nn.Dropout(dropout), - nn.Linear(64, output_dim)) - + classifier_base = nn.Sequential(nn.Linear(hidden_dim, 64), + nn.ReLU(), + nn.Dropout(dropout), + nn.Linear(64, output_dim)) if classifier_type == 'global': classifier = GlobalClassifier(classifier_base) elif classifier_type == 'last':