pytorch/caffe2
Neha Shah f3c00047ce Reset Optimizer counter while deserializing netWithBackwardOptions
Summary: Add ability to reset optimizer counter..

Test Plan: will wait for integration tests to run on diff.

Differential Revision: D27248286

fbshipit-source-id: a608df1bd61b64eb317c9ffd9cfdd804c5288f6d
2021-03-23 11:16:11 -07:00
..
contrib [caffe2] Fix caffe2 build with TensorRT support (#54322) 2021-03-22 13:19:08 -07:00
core fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
cuda_rtc
db
distributed Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
experiments Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
ideep Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
image Check kernel launches in caffe2/caffe2/image (#52173) 2021-02-12 10:11:41 -08:00
mobile [PyTorch Mobile] Add continuous build config for xplat/caffe2 2020-12-09 16:58:20 -08:00
mpi Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
observers Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
onnx Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
operators fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
opt Avoid of creating a copy of statusString every inference time (#53756) 2021-03-10 16:58:02 -08:00
perfkernels fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
predictor Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
proto [caffe2] Support deserializing tensors using alternate serialization formats (#53403) 2021-03-12 11:35:15 -08:00
python Reset Optimizer counter while deserializing netWithBackwardOptions 2021-03-23 11:16:11 -07:00
quantization fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
queue fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
serialize Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
sgd [caffe2 and pytorch] replace temp name of new sparse adagrad JIT'ed function in fbgemm (#52193) 2021-02-13 10:23:36 -08:00
share [PyTorch] Remove CAFFE2_FB_LIMITED_MOBILE_CAPABILITY (#50385) 2021-01-20 10:26:54 -08:00
test
transforms fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
utils fix comparison of narrow type with wide type in loop condition (#53951) 2021-03-22 16:40:35 -07:00
video Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
.clang-format
__init__.py
c2_aten_srcs.bzl
CMakeLists.txt Add cusolver potrf and potrfBatched to the backend of torch.cholesky decomposition (#53104) 2021-03-18 05:35:40 -07:00
README.md
release-notes.md Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
requirements.txt Remove references to typing module from setup.py (#47677) 2020-11-12 10:04:38 -08:00
VERSION_NUMBER

Caffe2

Jenkins Build Status

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.

Questions and Feedback

Please use GitHub issues (https://github.com/pytorch/pytorch/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai