facebookresearch/PoseDiffusion

Error executing job with overrides

zhuhu00 opened this issue · 2 comments

zhuhu00 commented

hi, jianyuan
This error occurred when I ran your code. Is this normal?
image

jytime commented

Hi,

Can you check what is the value of device when you run it? Usually it should be cuda or cpu, as:

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

If it works as expected, could you then check what would happen if you force all the tensors as .cuda() or .cpu()? It looks like the device of your machine is not found correctly.

zhuhu00 commented

Thanks, I forgot to modify it when i try to test on multi-gpu before. It's solved now.