Rohan Varma 79a40b22aa [Checkpoint] Make checkpoint_wrapper an nn.Module (#70164) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70164 Implement Alban's suggestion to make checkpoint_wrapper an nn.Module instead of patching the forward pass, which is too hacky. ghstack-source-id: 146011215 Test Plan: IC Reviewed By: mrshenli Differential Revision: D33214696 fbshipit-source-id: dc4b3e928d66fbde828ab60d90b314a8048ff7a2		2021-12-20 13:22:28 -08:00
..
_checkpoint	[Checkpoint] Make checkpoint_wrapper an nn.Module (#70164)	2021-12-20 13:22:28 -08:00
ddp_comm_hooks	[DDP Comm Hook] Create a noop hook for performance debugging (#64344)	2021-09-01 17:36:22 -07:00
model_averaging	[Reland] [Model Averaging] Simplify PostLocalSGD Optimizer API (#65197)	2021-09-17 10:31:58 -07:00
quantization	[BE] minor improvement to dist quantization (#67401)	2021-11-21 23:31:59 -08:00
__init__.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00
join.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00