Skip to content

Commit 494571e

Browse files
committed
update default epochs to 1
1 parent 69e53a7 commit 494571e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

eval_protocol/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,7 @@ def parse_args(args=None):
395395
rft_parser.add_argument("--base-model", help="Base model resource id")
396396
rft_parser.add_argument("--warm-start-from", help="Addon model to warm start from")
397397
rft_parser.add_argument("--output-model", help="Output model id (defaults from evaluator)")
398-
rft_parser.add_argument("--epochs", type=int, default=8)
398+
rft_parser.add_argument("--epochs", type=int, default=1)
399399
rft_parser.add_argument("--batch-size", type=int, default=128000)
400400
rft_parser.add_argument("--learning-rate", type=float, default=3e-5)
401401
rft_parser.add_argument("--max-context-length", type=int, default=65536)

0 commit comments

Comments
 (0)