|
|
@ -12,7 +12,7 @@ if __name__ == '__main__': |
|
|
|
# train |
|
|
|
parser.add_argument('--lr', type=float, default=2e-4) |
|
|
|
parser.add_argument('--weight_decay', type=float, default=2.5e-5) |
|
|
|
parser.add_argument('--epochs', type=int, default=512) |
|
|
|
parser.add_argument('--epochs', type=int, default=100) |
|
|
|
parser.add_argument('--batch_size', type=int, default=128) |
|
|
|
parser.add_argument('--betas', type=float, nargs='+', default=(.5, .999)) |
|
|
|
# misc |
|
|
|