RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

yccyenchicheng / SDFusion

MIT License

406 stars 34 forks source link

RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu) #46

Open SteelDominic opened 8 months ago

SteelDominic commented 8 months ago

When i try to train train_sdfusion_snet,I met this problem.I have trained the vqave model on the ShapeNet model successfully.How can i fix it?

Xiaofei-CN commented 3 months ago

I find the same error. I check self.logvar and find it doesn't in cuda self.logvar = torch.full(fill_value=logvar_init, size=(self.num_timesteps,)).to(self.device)