Add validation loss

2026-04-09 06:45:09 +00:00 · 2023-11-05 12:35:46 -05:00
parent 95ae56bd22
commit 5b19bda85c
2 changed files with 120 additions and 1 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -4736,6 +4736,10 @@ class collator_class:
        else:
            dataset = self.dataset

+        # If we split a dataset we will get a Subset
+        if type(dataset) is torch.utils.data.Subset:
+            dataset = dataset.dataset
+
        # set epoch and step
        dataset.set_current_epoch(self.current_epoch.value)
        dataset.set_current_step(self.current_step.value)