code update

2025-05-19 22:54:53 +02:00
parent 58305effdb
commit cb7896900b
7 changed files with 238 additions and 51 deletions
@@ -55,6 +55,11 @@ def simple_x_y_collate(batch):
        collated_x.append(np.concatenate(current_x, axis=1))
        collated_y.append(np.concatenate(current_y, axis=1))

+    # convert to numpy arrays
+    collated_x = np.array(collated_x)
+    collated_y = np.array(collated_y)
+
+    # convert to torch tensors
    return torch.tensor(collated_x, dtype=torch.float32), torch.tensor(collated_y, dtype=torch.float32)