Train more epochs for GitHub actions.

This commit is contained in:
Fangjun Kuang 2021-08-23 08:30:39 +08:00
parent 3ffcd95086
commit 6617d5828d

View File

@ -59,6 +59,13 @@ def get_parser():
help="Should various information be logged in tensorboard.", help="Should various information be logged in tensorboard.",
) )
parser.add_argument(
"--num-epochs",
type=int,
default=50,
help="Number of epochs to train.",
)
return parser return parser