About 3D dataset

[jiashuow]

Thank you so much for sharing this work!

I'm using your code on a 3D dataset (parse22). And I met a problem that I don't understand. It reports RuntimeError: Expected weight to be a vector of size equal to the number of channels in input, but got weight of shape [1] and input of shape [2, 4, 1, 64, 64, 64].

To run your model, I only changed or added the dataset-specific part in the src/configs folder and src/data_loader folder. Do you know why this error occurred? Thanks a lot!

I attached the whole repertory I used and the traceback and report error below.

vessel-flow-consistency-ssl.zip