search

IDEA-Research / HumanSD

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"
Apache License 2.0
271 stars 18 forks source link

'cuda' is not a valid DistributedType #24

Closed Kai-0515 closed 10 months ago

Kai-0515 commented 10 months ago

Hi when I train the humansd model, I met the 'cuda' is not a valid DistributedType' I set the GPU to 1 can not fix.

截屏2023-10-20 下午4 35 17
VicZlq commented 9 months ago

Hi when I train the humansd model, I met the 'cuda' is not a valid DistributedType' I set the GPU to 1 can not fix. 截屏2023-10-20 下午4 35 17

Hello, did you follow up on this issue? I had a similar problem when I reproduced it. I changed the 'cuda' to 'gpu' and the bug was solved, but then opt.train=False and I couldn't get to trainer.fit(model, data), have you encountered this?