Train 3D Networks with Pytorch
Hi all, Thank you in advance.
ltnghia
apaszke
Thanks. Do you support Deconv3D?
ltnghia
apaszke
The return 4D elements is (c,d,h,w) ?
YunnFeng
The return is a tuple of 5 elements (N, c, d, h, w).
Ismail_Elezi
You need to return 4D elements from the dataset, so they will be concatenated into 5D batches. Then, we only support batch mode in nn, so you’ll be using 5D tensors throught the network.
apaszke
YunnFeng
apaszke