set the default number of epochs to 20

nanoporetech · Sep 4, 2020 · 9746bcc · 9746bcc
1 parent 28b1e66
commit 9746bcc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bonito/train.py b/bonito/train.py
@@ -111,7 +111,7 @@ def argparser():
     parser.add_argument("--device", default="cuda")
     parser.add_argument("--lr", default=1e-3, type=float)
     parser.add_argument("--seed", default=25, type=int)
-    parser.add_argument("--epochs", default=400, type=int)
+    parser.add_argument("--epochs", default=20, type=int)
     parser.add_argument("--batch", default=32, type=int)
     parser.add_argument("--chunks", default=2000000, type=int)
     parser.add_argument("--validation_split", default=0.97, type=float)