Thanks for your code first but there are some problems for me to run it.
In main.py line 121 to line 123, you used Loss_label1, Loss_fake1, Loss_unlabel1 while you only imported Loss_label, Loss_fake, Loss_unlabel in the losses.py . Are Loss_label1 and Loss_label mean the same loss? If so, I met RuntimeError: CUDA error: device-side assert triggered while compute the loss. Have you met an error like this?
Hi Gengyan, could you please give me the .h5 dataset link or info that you have used in this experiment. I want to make custom dataloader based on that info. Thanks in advance.