mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Summary: predictor_exporter copies the original predict_net's op, external_input and external_output fields, but ignores the type field. This is reasonable as the train net would generally have 'dag' type and copying that for inference may not be applicable. It's good to have a way to specify the net type nevertheless to run DAGNet for inference. This diff adds a field in predictor_exporter to do that. Reviewed By: akyrola Differential Revision: D5122354 fbshipit-source-id: 0e3cc417128db903c71515135c9e3b87620ae21e |
||
|---|---|---|
| .. | ||
| mobile_exporter_test.py | ||
| mobile_exporter.py | ||
| predictor_exporter_test.py | ||
| predictor_exporter.py | ||
| predictor_py_utils.py | ||
| serde.py | ||