"None value not supported" when i use the code in multi_modal segmentation

[miss1997yuan]

hi,can you help me ,i have met the "None value not supported" when i try to use the code in medical
pet-ct image segmentation ,i am embrassed to find the problem ,the special point is the input is two
picture for ct and pet image,and the output segmentation logits is not as size [batch,num_classes],it is about the size[batch,h,w,num_classes] for i try to calculate the all pixels' grad in the sapce of image ad [h,w]
have no idea of is it possible to use the technology in the segmentation task rather than classification ,would you mind giving me some suggestion ?

[marcoancona]

I am afraid this is not supported by DeepExplain. The output is expected to be of shape [batch, n_classes]. Also, how would that work for image segmentation? Would you generate one heatmap for each of your h*w*num_classes outputs? Seems difficult to handle to me.