From b491a817088790219e052b86173e128c55b597f8 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 23 Dec 2022 21:53:46 +0100 Subject: num_class_images is now class images per train image --- train_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 55daa35..e272b5d 100644 --- a/train_ti.py +++ b/train_ti.py @@ -83,7 +83,7 @@ def parse_args(): parser.add_argument( "--num_class_images", type=int, - default=400, + default=1, help="How many class images to generate." ) parser.add_argument( -- cgit v1.2.3-54-g00ecf