diff --git a/train_enc.py b/train_enc.py index 1dcaf2d..1469c75 100644 --- a/train_enc.py +++ b/train_enc.py @@ -20,7 +20,7 @@ help='Number of epochs to train.') parser.add_argument('--batch-size', type=int, default=128, help='Number of samples per batch.') -parser.add_argument('--lr', type=float, default=0.01, +parser.add_argument('--lr', type=float, default=0.0005, help='Initial learning rate.') parser.add_argument('--hidden', type=int, default=512, help='Number of hidden units.')