pytorch/caffe2/python/predictor
Aapo Kyrola 46a95cf420 Allow specifying device to load_from_db()
Summary: A quite common problem is that it is hard to load blobs with pe.load_from_db to a specific device. One must set the device options of the returned init_net and predict_init_net, which is quite magical. So I made load_from_db() able to set these device options automatically, based on device scope or device_option parameter. Added an unit test.

Reviewed By: asaadaldien

Differential Revision: D5249202

fbshipit-source-id: 7b9d91476cb8d1b0ec0d9772e50b9148b8b184fa
2017-06-14 14:32:24 -07:00
..
mobile_exporter_test.py Added mobile exporter 2017-05-24 11:36:44 -07:00
mobile_exporter.py Added mobile exporter 2017-05-24 11:36:44 -07:00
predictor_exporter_test.py Allow specifying device to load_from_db() 2017-06-14 14:32:24 -07:00
predictor_exporter.py Allow specifying device to load_from_db() 2017-06-14 14:32:24 -07:00
predictor_py_utils.py Allow specifying net type in predictor_exporter 2017-05-24 11:46:27 -07:00
serde.py Re-apply #266 2017-04-25 21:17:04 -07:00