mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
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 |
||
|---|---|---|
| .. | ||
| mobile_exporter_test.py | ||
| mobile_exporter.py | ||
| predictor_exporter_test.py | ||
| predictor_exporter.py | ||
| predictor_py_utils.py | ||
| serde.py | ||