JJH

Results 2 issues of JJH

Hello! ![image](https://user-images.githubusercontent.com/41336761/161049644-b26918af-909c-4c25-994d-5e92a7aea8f4.png) when i trained with my own dataset, some error occured, i do not know how to fix it. my task is multi classes segmentation

Thanks for your great job. I want to test the code on the 3D data, Could you please give me some advice on how to change the code?