From 5ee7f52ecad439fe1b7e1f887d84eaf68f41f56a Mon Sep 17 00:00:00 2001 From: wogong Date: Wed, 4 Sep 2019 22:01:39 +0800 Subject: [PATCH] add l2 norm. --- core/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/train.py b/core/train.py index 6ad0b6f..8da0c46 100644 --- a/core/train.py +++ b/core/train.py @@ -21,7 +21,7 @@ def train_dann(model, params, src_data_loader, tgt_data_loader, tgt_data_loader_ if not params.finetune_flag: print("training non-office task") - optimizer = optim.SGD(model.parameters(), lr=params.lr, momentum=params.momentum) + optimizer = optim.SGD(model.parameters(), lr=params.lr, momentum=params.momentum, weight_decay=params.weight_decay) else: print("training office task") parameter_list = [{