From b448835960293dd916e0c83741fb01c89fca294b Mon Sep 17 00:00:00 2001 From: fazilaltinel Date: Thu, 10 Dec 2020 15:40:59 +0300 Subject: [PATCH] Change num of epochs --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index ac43bee..19c7715 100644 --- a/main.py +++ b/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