Explorar o código

Fix data split ratio

Lengyue %!s(int64=2) %!d(string=hai) anos
pai
achega
bce85f5c35
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  1. 2 2
      tools/vqgan/create_train_split.py

+ 2 - 2
tools/vqgan/create_train_split.py

@@ -26,8 +26,8 @@ def main(root, val_ratio, val_count, filelist):
     Random(42).shuffle(files)
 
     if val_count is None and val_ratio is None:
-        logger.info("Validation ratio and count not specified, using max(20%, 100)")
-        val_size = max(1, math.ceil(len(files) * 0.2))
+        logger.info("Validation ratio and count not specified, using min(20%, 100)")
+        val_size = min(1, math.ceil(len(files) * 0.2))
     elif val_count is not None and val_ratio is not None:
         logger.error("Cannot specify both val_count and val_ratio")
         return