CU-Net
CU-Net copied to clipboard
Encounter Segmentation fault when running loss.backword()
HI, thanks for your impressive code. When I used your code to train a native dense unet, loss.backword caused Segmentation fault, while the forward is ok. Can you help me, Thanks very much!
Please make sure your Pytorch version matches the corresponding script. Script names with "prev-version" require Pytorch v0.1.12.