commit | 152d43940048c3c366a34df830097302bbbea0ff | [log] [tgz] |
---|---|---|
author | Viswanath Sivakumar <viswanath@fb.com> | Wed May 24 11:39:49 2017 -0700 |
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed May 24 11:46:27 2017 -0700 |
tree | 0ff82043db9e18001a89f822a700aed4b4dbf7f1 | |
parent | 03503140fd75dc9c53c90a09d16c56a2a1d36fcd [diff] |
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
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the BSD 2-Clause license.
Detailed build matrix (hit refresh if you see icons not showing up due to heroku):
Target | Status |
---|---|
Linux | |
Mac (CPU) | |
Android | |
iOS | |
Linux + MKL | |
Windows |