Add to device here

[zihaoz96]

https://github.com/qanastek/HugsVision/blob/d93cdd15da6fc9fe7f3053ca6e6d4196ef758327/hugsvision/inference/TorchVisionClassifierInference.py#L46

[qanastek]

RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu

[qanastek]

The issue is related to the DataParallel layer at the end of the initialization block for the TorchVision models. Thus, we need to remove the layer to make it working on CPU devices.

Note: Will be merged in the version 0.72

Code edited:

• https://github.com/qanastek/HugsVision/blob/d93cdd15da6fc9fe7f3053ca6e6d4196ef758327/hugsvision/nnet/TorchVisionClassifierTrainer.py#L117
• https://github.com/qanastek/HugsVision/blob/d93cdd15da6fc9fe7f3053ca6e6d4196ef758327/hugsvision/nnet/TorchVisionClassifierTrainer.py#L146

Sources:

• https://discuss.pytorch.org/t/does-dataparallel-matters-in-cpu-mode/7587

[qanastek]

Close with the push: 8529361