pytorch/caffe2/python/predictor
Viswanath Sivakumar 152d439400 Allow specifying net type in predictor_exporter
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
2017-05-24 11:46:27 -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 net type in predictor_exporter 2017-05-24 11:46:27 -07:00
predictor_exporter.py Allow specifying net type in predictor_exporter 2017-05-24 11:46:27 -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