(pytorch issue) torch.load not implemented

artyom-beilis / pytorch_dlprim

DLPrimitives/OpenCL out of tree backend for pytorch

http://blog.dlprimitives.org/

MIT License

264 stars 17 forks source link

Open zougloub opened 1 year ago

zougloub commented 1 year ago

Cause: deserialization code not implemented (serialization.py / default_restore_location()).

Workaround: load model to CPU, and transfer to ocl device.