Dirk Groeneveld 75945d54f7 Properly propagates checkpoint wrapper args and kwargs (#99791) ... It looks like passing `*args` and `**kwargs` to `checkpoint_wrapper()` does not work because someone forgot some `*`s. This adds them back in. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99791 Approved by: https://github.com/awgu		2023-05-03 23:19:21 +00:00
..
__init__.py
checkpoint_wrapper.py	Properly propagates checkpoint wrapper args and kwargs (#99791)	2023-05-03 23:19:21 +00:00