3D CUT for medical images

[CasperDL2805]

Hi!

I have altered your code to work with 3D images, however, I do not see a gradual decrease in NCE loss (see image). I could not find any images of your loss, and I know it is highly dependent on the data etc - but do you see a gradual decrease? In that case, it may indicate that I have done something wrong... I have trained the CycleGAN with the same data, and here I see a gradual decrease in both Cycle loss and Identity loss.

Thank you in advance!