From 4878edd0cf696d8f8a82f800faf29b7e7c77edb3 Mon Sep 17 00:00:00 2001 From: OleehyO Date: Mon, 13 Jan 2025 12:48:21 +0000 Subject: [PATCH] fix: correct do_validation argument parsing --- finetune/schemas/args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune/schemas/args.py b/finetune/schemas/args.py index be7fd6f..a01ea18 100644 --- a/finetune/schemas/args.py +++ b/finetune/schemas/args.py @@ -229,7 +229,7 @@ class Args(BaseModel): parser.add_argument("--resume_from_checkpoint", type=str, default=None) # Validation - parser.add_argument("--do_validation", type=bool, default=False) + parser.add_argument("--do_validation", type=lambda x: x.lower() == 'true', default=False) parser.add_argument("--validation_steps", type=int, default=None) parser.add_argument("--validation_dir", type=str, default=None) parser.add_argument("--validation_prompts", type=str, default=None)