In that case your channel dimension is missing and could be added via:
x = x.unsqueeze(1)
Also, since you are dealing with volumetric data, you would have to use the nn.*3d
modules, such as nn.Conv3d
.
In that case your channel dimension is missing and could be added via:
x = x.unsqueeze(1)
Also, since you are dealing with volumetric data, you would have to use the nn.*3d
modules, such as nn.Conv3d
.