yin huang

Results 1 comments of yin huang

batch_size设置大于1时报错,如下,batch_size为4 File "train.py", line 326, in train model.optimize_parameters() File "/public/home/hatmore/project/CWSAM-master/models/sam.py", line 289, in optimize_parameters self.backward_G() # calculate graidents for G self.backward_G() # calculate graidents for G File "/public/home/hatmore/project/CWSAM-master/models/sam.py", line 280,...