Browse Source

Delete unnecessary comment lines

master
Fazil Altinel 4 years ago
parent
commit
e1d1039533
  1. 13
      core/train.py

13
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) optimizer = optim.SGD(model.parameters(), lr=params.lr, momentum=params.momentum, weight_decay=params.weight_decay)
else: else:
print("training office task") 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 = [{ parameter_list = [{
"params": model.feature_extractor.parameters(), "params": model.feature_extractor.parameters(),
"lr": 0.0001 "lr": 0.0001

Loading…
Cancel
Save