AssertionError: Invalid device id

snap-research / CAT

[CVPR 2021] Teachers Do More Than Teach: Compressing Image-to-Image Models (CAT)

https://dejqk.github.io/GAN_CAT/

Other

179 stars 20 forks source link

AssertionError: Invalid device id #7

Closed XUEXI-CL closed 2 years ago

XUEXI-CL commented 2 years ago

Hello, I encountered errors in model training. What should I do? Do you have any suggestions? Thank you very much！ “AssertionError: Invalid device id”

alanspike commented 2 years ago

Hey, I guess the gpu ids should be adjusted according to your computation resources. For example, changing the gpu ids here.