So on line 163 in https://github.com/sweetcocoa/ddsp-pytorch/blob/5d0247e8b1c108662f7ac2507625103471d27e49/train/trainer/trainer.py they do that map to self.device, and self.device is earlier initialized to the output of get_model_device function above. Firstly, I’m not sure why the model being set to cuda:0 is a problem since if i remember correctly the model actually stays on the default device no? Even if this is wrong, i tried removing the map_location argument and got the same error…
However! I bet line 316-327 in the same link above in trainer.py really matters. They send the batch to the device, and I’m not sure what I shoudl replace to self.device with