Initial Commit

2023-04-07 09:44:12 +00:00
parent 42d655a451
commit c49645d7bc
13 changed files with 423 additions and 128 deletions
--- a/datasets/czech_slr_dataset.py
+++ b/datasets/czech_slr_dataset.py
@@ -30,6 +30,7 @@ class CzechSLRDataset(torch_data.Dataset):
        self.data = data
        self.labels = labels
        self.targets = list(labels)
+
        self.num_labels = num_labels
        self.transform = transform