In main.py the help strings for these two arguments seem to be flipped:
parser.add_argument('--n_batch_per_epoch', default=2000, type=int, help="Number of training epochs")
parser.add_argument('--nb_epoch', default=400, type=int, help="Number of batches per epoch")
In main.py the help strings for these two arguments seem to be flipped: