Proper interaction of save/load and cpu/gpu transfer

GateNLP / gate-lf-pytorch-json

PyTorch wrapper for the LearningFramework GATE plugin

Apache License 2.0

1 stars 2 forks source link

Proper interaction of save/load and cpu/gpu transfer #13

Open johann-petrak opened 6 years ago

johann-petrak commented 6 years ago

Make sure all data is transferred to the CPU before saving, then after loading make sure the GPU status is determined and the data transferred to the GPU as needed or not.