Slice thickness is the input channels?

[suoranxiu]

Apologizing for bothering you. I'm running this code on my own lymphoma datasets. And there are some problems when initializing parameters.

slice_thickness = 3

Slice thickness is the input channels?

class RSTN(nn.Module): def __init__(self, crop_margin=20, crop_prob=0.5,crop_sample_batch=1, n_class=3, TEST=None):

"n_class" is the output channels of the model?

[twni2016]

Yes, slice_thickness and n_class should be the same number (here 3) because input and output channels should be the same.

[suoranxiu]

Thank you so much!