Browse Source

更新 'train_cload.py'

master
lwj 1 month ago
parent
commit
8a9d480ae7
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      train_cload.py

+ 3
- 2
train_cload.py View File

@@ -303,8 +303,9 @@ def main(args):
# learning_rates.append(args.lr*(args.gamma**(i/2)))
# lr = nn.piecewise_constant_lr(milestone, learning_rates)

iter_lr = poly_lr(args.lr, 7120 * 2, 7120 * 100)
optimizier = nn.Adam(net.trainable_params(), learning_rate=iter_lr, beta1=0.9, beta2=args.beta2)
# iter_lr = poly_lr(args.lr, 7120 * 2, 7120 * 100)
# optimizier = nn.SGD(net.trainable_params())
optimizier = nn.Adam(net.trainable_params(), learning_rate=iter_lr, beta1=0.9, beta2=args.beta2, weight_decay=0.01)


# Save model config


Loading…
Cancel
Save