Tensors and Dynamic neural networks in Python with strong GPU acceleration
Find a file
Yangqing Jia bcea409c82 sync
2016-07-28 15:06:43 -07:00
caffe bring up caffe.proto to master 2016-07-28 09:55:49 -07:00
caffe2 sync 2016-07-28 15:06:43 -07:00
docs chunky sync 2016-05-13 14:43:48 -07:00
third_party changes to make c2 build. 2016-07-21 16:39:08 -07:00
.Doxyfile Adding optional Eigen code. Added a switch USE_SYSTEM_EIGEN in Env. Misc changes. 2015-10-18 16:55:24 -07:00
.gitignore Adding optional Eigen code. Added a switch USE_SYSTEM_EIGEN in Env. Misc changes. 2015-10-18 16:55:24 -07:00
.gitmodules changes to make c2 build. 2016-07-21 16:39:08 -07:00
build.py changes to make c2 build. 2016-07-21 16:39:08 -07:00
build_android.py Android build improvements 2016-07-26 12:48:53 -07:00
build_android_prepare.py Add a simple script to help build android. 2016-07-26 10:40:37 -07:00
LICENSE A memory pool implementation based on cnmem. Added cnmem license to LICENSE. 2015-09-03 20:55:50 -07:00
Makefile move a bunch of things 2016-03-08 15:15:19 -08:00
README.md Update README.md 2016-05-16 21:42:12 -07:00

Caffe2

Caffe2 is a deep learning framework made with expression, speed, and modularity in mind. It is an experimental refactoring of Caffe, and allows a more flexible way to organize computation.

Read the installation instructions for installation details.

What Caffe2 is and what it is not

Caffe2 started as an attempt to improve the design of caffe, because we have learned quite a lot since the 2 years of running and using caffe. It is also aiming to make Caffe more general, not only focusing on vision but applicable to all kinds of machine learning tasks.

Some highlights of the Caffe refactoring idea can also be found at the slides from the CVPR caffe tutorial.

Note that this is not meant for a complete replacement for Caffe right now. You are more than welcome to try it and we appreciate much if you could contribute back, but a lot of things may still be missing (e.g. CuDNN) so expect a few bumps as an early adopter. Caffe is still stably supported and for a production environment you may want to continue using it.

License and Citation

Caffe2 is released under the BSD 2-Clause license.