Open silky1708 opened 1 month ago
first move self.logvar to self.device, and then index using t (line 1030, ddpm.py) the current verison of the code gives the following error:
self.logvar
self.device
t
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
remove extraneous argument outputs from on_train_epoch_end in main.py the current version of the code gives the following error:
outputs
on_train_epoch_end
main.py
TypeError: on_train_epoch_end() missing 1 required positional argument: 'outputs'
first move
self.logvar
toself.device
, and then index usingt
(line 1030, ddpm.py) the current verison of the code gives the following error:remove extraneous argument
outputs
fromon_train_epoch_end
inmain.py
the current version of the code gives the following error: