From 7ce728b7ea9cfe6b6dc7d05826c1bf64eec5aacb Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Mon, 9 Jan 2023 10:57:05 +0100
Subject: Enable buckets for validation, fixed vaildation repeat arg

---
 train_ti.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

(limited to 'train_ti.py')

diff --git a/train_ti.py b/train_ti.py
index 7784d04..df8d443 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -383,7 +383,7 @@ def parse_args():
     parser.add_argument(
         "--valid_set_repeat",
         type=int,
-        default=None,
+        default=1,
         help="Times the images in the validation dataset are repeated."
     )
     parser.add_argument(
@@ -477,9 +477,6 @@ def parse_args():
     if isinstance(args.exclude_collections, str):
         args.exclude_collections = [args.exclude_collections]
 
-    if args.valid_set_repeat is None:
-        args.valid_set_repeat = args.train_batch_size
-
     if args.output_dir is None:
         raise ValueError("You must specify --output_dir")
 
-- 
cgit v1.2.3-54-g00ecf