minimum value for validation dataset

acids-ircam · May 7, 2022 · e26bdfd · e26bdfd
1 parent 16ea4a9
commit e26bdfd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_rave.py b/train_rave.py
@@ -106,7 +106,7 @@ class args(Config):
         ]),
     )
 
-    val = (2 * len(dataset)) // 100
+    val = max((2 * len(dataset)) // 100, 1)
     train = len(dataset) - val
     train, val = random_split(
         dataset,