search

facebookresearch / DomainBed

DomainBed is a suite to test domain generalization algorithms
MIT License
1.42k stars 298 forks source link

Fix running on CPU #30

Closed daysm closed 4 years ago

daysm commented 4 years ago

Some algorithms did not work on CPU since they tried to create tensors on the GPU. For those tensors, I replaced .cuda() with .to(device).