I was trying to use the following code segment to leverage a pre-trained model.
ENCODER_NAME = \'se_resnext50_32x4d\' class HuBMAPModel(nn.Module): def _