If you want to use size of 64x64x64 volume to train network, you can random crop volume from initial size of 240x240x155. Then you should make getitem() function output 1depthheight*depth and use DataLoader to warp it.