Browse Source

更新 'train_cload.py'

master
lwj 1 month ago
parent
commit
4559e73ef2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      train_cload.py

+ 1
- 1
train_cload.py View File

@@ -305,7 +305,7 @@ def main(args):

# 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)
optimizier = nn.Adam(net.trainable_params(), learning_rate=args.lr, beta1=0.9, beta2=args.beta2, weight_decay=0.01)


# Save model config


Loading…
Cancel
Save