Merge pull request #243 from rgxb2807/fine_transformer_valid_ds_fix

Bugfix - Fixing validation dataset variable on FineTransformerTrainer
lucidrains · Oct 30, 2023 · ea973a0 · ea973a0
2 parents 9a0cdbe + ea98e13
commit ea973a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/audiolm_pytorch/trainer.py b/audiolm_pytorch/trainer.py
@@ -1207,9 +1207,9 @@ def __init__(
 
         # split for validation
 
-        self.valid_dataset = valid_dataset
+        self.valid_ds = valid_dataset
 
-        if not exists(self.valid_dataset):
+        if not exists(self.valid_ds):
             if valid_frac > 0:
                 train_size = int((1 - valid_frac) * len(self.ds))
                 valid_size = len(self.ds) - train_size