saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/python/predictor
History
Aapo Kyrola 6384bae29b call save_to_db in CPUContext + fix a typo in data_parallel_model. 
Summary:
If Predictor Exporter save_to_db is called in CUDAContext, a failure occurs since the following FeedBlob() tries to store a string (meta data), but for CUDA blobs we assume they are tensors.
  + fix a typo in data_parallel_model that I bumped on.

Reviewed By: asaadaldien

Differential Revision: D5099837

fbshipit-source-id: 69d01b35a9a1816bf083f13d8a6ce88e1f5aecb7
2017-05-19 18:25:00 -07:00
..
predictor_exporter.py call save_to_db in CPUContext + fix a typo in data_parallel_model. 2017-05-19 18:25:00 -07:00
predictor_exporter_test.py Predictor exporter open-sourcing 2017-04-06 10:01:42 -07:00
predictor_py_utils.py Re-apply #266 2017-04-25 21:17:04 -07:00
serde.py Re-apply #266 2017-04-25 21:17:04 -07:00