| commit | 7bf4c0e0fb396faa8e3cbacfe2d9d8d5a776d564 | [log] [tgz] |
|---|---|---|
| author | Aapo Kyrola <akyrola@fb.com> | Wed Jun 14 22:31:04 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed Jun 14 22:32:29 2017 -0700 |
| tree | e45c3a00869aed4c4386480fdf3faee7856441b0 | |
| parent | 2ec294a8bbcab1585eb2f8d197f08c1f78f5d2a0 [diff] |
support RNNs in ExtractPredictorNet Summary: We need to support RNNs explicitly in ExtractPredictorNet, because they store sub-nets as strings in special arguments. When netdef argument arrive, we can generalize this a bit. Added a test under rnn_cell_test to test that extracting an LSTM predictor net works correctly and sets the device option properly for the step net ops. Reviewed By: yqwangustc Differential Revision: D5236334 fbshipit-source-id: cd653427f8c440a14d94195a532d18276f94749a
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.
Caffe2 Bay Area Meetup at NVIDIA, May 31 6-8:30, Santa Clara, CA: https://www.meetup.com/Caffe2-Bay-Area/events/239836290/
Caffe2 Community Facebook Group: join to ask questions, talk to other users, and keep informed of important Caffe2 updates.
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.
| Target | Status |
|---|---|
| Linux | |
| Android | |
| iOS | |
| Linux + MKL |