Upload folder using huggingface_hub

train_and_test.py

@@ -121,6 +121,8 @@ print('Loading training files into RAM...')
 train_wo, train_w, train_lbl = [], [], []
 for fp in trainpath:
     wo, w, lbl = load_data(fp)
+    if lbl.dim() == 3:
+        lbl = lbl.unsqueeze(1)
     train_wo.append(wo)
     train_w.append(w)
     train_lbl.append(lbl)
@@ -143,6 +145,8 @@ print('Loading testing files into RAM...')
 test_wo, test_w, test_lbl = [], [], []
 for fp in testpath:
     wo, w, lbl = load_data(fp)
+    if lbl.dim() == 3:
+        lbl = lbl.unsqueeze(1)
     test_wo.append(wo)
     test_w.append(w)
     test_lbl.append(lbl)