AuroraYXL

Loss does not fall

3

I have two questions: 1. When training with multiple GPUs, it seems that self.genbasis cannot be copied to all GPUs. 2. My loss only decreased in the first 5K iterations,...

repeat_interleave

1

After I reset the environment according to "Requirements", this line of code was misaligned, and when I changed "axis" to "dim" the program worked fine. 

Arguments are called before they are defined

1

img1 img2 are called before they are defined. data1,data2->img1 img2?