From e1d1039533049c70a6efec9c174214d2525e6a0e Mon Sep 17 00:00:00 2001 From: fazilaltinel Date: Tue, 22 Dec 2020 10:27:42 +0300 Subject: [PATCH] Delete unnecessary comment lines --- core/train.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/core/train.py b/core/train.py index 9c88076..217fc07 100644 --- a/core/train.py +++ b/core/train.py @@ -126,19 +126,6 @@ def train_dann(model, params, src_data_loader, tgt_data_loader, tgt_data_loader_ optimizer = optim.SGD(model.parameters(), lr=params.lr, momentum=params.momentum, weight_decay=params.weight_decay) else: print("training office task") - # parameter_list = [{ - # "params": model.features.parameters(), - # "lr": 0.001 - # }, { - # "params": model.fc.parameters(), - # "lr": 0.001 - # }, { - # "params": model.bottleneck.parameters() - # }, { - # "params": model.classifier.parameters() - # }, { - # "params": model.discriminator.parameters() - # }] parameter_list = [{ "params": model.feature_extractor.parameters(), "lr": 0.0001