pengyan510 / nlp-paper-implementation

33 stars 12 forks source link

Config yaml default device #2

Open Jeff-Winchell opened 11 months ago

Jeff-Winchell commented 11 months ago

The default yaml file in this repository has the line device: cpu

cpu is not a string but something else, which I can't seem to find so it breaks the code when you pass it to the model.to() method.

Jeff-Winchell commented 11 months ago

it should be a string in the yaml file. Possible values seem to be one of these strings: cpu, cuda, ipu, xpu, mkldnn, opengl, opencl, ideep, hip, ve, fpga, ort, xla, lazy, vulkan, mps, meta, hpu, privateuseone

Viile1 commented 5 months ago

fpga ne alaka be,