fmdmm

Results 2 issues of fmdmm

看到:“batch_outputs[0].cpu().numpy()”这块,但是单gpu为啥就没有问题,想不通呀 ![image](https://user-images.githubusercontent.com/102396016/179695053-71561b1b-f40c-4417-9a55-5eba3dc1c42e.png) 具体报错如下: Training: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████| 640/640 [08:48