You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Boyuan Yao de1e716dc4 [fx] Add activation checkpoint solver rotor (#1496) ... * [fx] fix defining ckpt functions inside forward * [fx] Modify activation checkpoint codegen and add ColoGraphModule * [fx] some modification * some modifications * some modifications * some modifications * some modifications * some code modifications * [automatic_parallel] ckpt solver rotor * [fx] add ckpt_solver_rotor * [fx] modification * code refactor * code refactor		2 years ago
..
test_ckpt_torchvision.py	[fx] Add activation checkpoint solver rotor (#1496)	2 years ago