| commit | 60aa8c793dfeec46bea211c2fbb167c7218197bf | [log] [tgz] |
|---|---|---|
| author | Alexander Sidorov <asidorov566@gmail.com> | Wed Mar 07 16:41:22 2018 -0800 |
| committer | GitHub <noreply@github.com> | Wed Mar 07 16:41:22 2018 -0800 |
| tree | 74ccf1c3d8e03e80db2b4d9ae372f7a2d5ae5d41 | |
| parent | 957ddb54d69928ed2d0d200e6d0f05eb116f0e16 [diff] |
Update caffe2 from facebook (#2178) * [C2] Don't crash kernel in case of invalid shapes for ConcatOp Enforce correctness of the shapes for input tensors so we won't access invalid index. * [Caffe2] Add analytical performance counters to Dynolog Initial diff for counting analytical flops and memory writes for C2 operators. * BBoxTransform op: Handle RoIs from multiple images per batch BBoxTransform op used during typical Faster-RCNN inference operates only on RoIs from a single image (no batching). Adding support to handle that with an optional output blob containing the batch splits (i.e., the number of RoIs belonging to each item in the batch). The code is perfectly backward compatible and shouldn't break any existing models.. * [mkl] Make MKL-DNN cooperate with memongered nets C2's MKL-DNN implementation caches input dims and reuses intermediate and output buffers across net runs, which prevents memonger from being used. This may not always be useful since input dims may vary widely in many cases and we'll end up reallocating anyway. Added an option to force reallocation when memonger is used. * [oncall] fix batch gather ops for empty input still need to bisect for the breaking change, but this shall fix the case for empty input. the error logging is like: https://interncache-ftw.fbcdn.net/t49.3276-7/23938497_293562711176943_6500112636590424064_n.txt?_nc_log=1 @[557759185:raychen] can you help to subscribe oncall from ads side. this may affect the Sigrid online trainer. * optimize BatchOneHotOp We want to iterate in row-major as opposed to column-major for better locality. * Supported exporting model with int blobs. Supported exporting model with int blobs. Needed by condensenet. * BoxWithNMSLimit op: Handle boxes from mutiple images per batch Similar to D7135360. Added support for multiple images per batch in the op. Takes an optional additional input "batch_splits" as output by BBoxTransform op, and returns new batch_splits after applying NMS and filtering. Otherwise, backward compatibility is maintained.
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 Apache 2.0 license. See the NOTICE file for details.