Skip to content
forked from pytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

License

Notifications You must be signed in to change notification settings

wellneck/pytorch

Repository files navigation

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.

About

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Resources

License

Code of conduct

Security policy

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C++ 50.5%
  • Python 38.1%
  • Cuda 4.2%
  • C 3.2%
  • Objective-C++ 1.3%
  • CMake 0.9%
  • Other 1.8%