Browse Source

Change num of epochs

master
Fazil Altinel 4 years ago
parent
commit
b448835960
  1. 2
      main.py

2
main.py

@ -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

Loading…
Cancel
Save