extend epoch length to 150
@@ -95,7 +95,7 @@ class Session():
def add_args(parser: argparse.ArgumentParser):
parser.add_argument('--session.learning_rate', default=0.01, type=float, help='Training initial learning rate.')
parser.add_argument('--session.momentum', default=0.98, type=float, help='Training initial momentum for SGD.')
- parser.add_argument('--session.epoch_length', default=10, type=int, help='Iterations of training per epoch')
+ parser.add_argument('--session.epoch_length', default=150, type=int, help='Iterations of training per epoch')
parser.add_argument('--session.batch_size_train', default=1, type=int, help='Training batch size.')
parser.add_argument('--session.sync_interval', default=100, type=int, help='Batches before we sync with chain and emit new weights.')
parser.add_argument('--session.log_interval', default=10, type=int, help='Batches before we log session info.')
