fix: torch.checkpoint() incorrectly wraps single forward step in original codebase.#274
Open
COLAZERO2 wants to merge 1 commit into
Open
fix: torch.checkpoint() incorrectly wraps single forward step in original codebase.#274COLAZERO2 wants to merge 1 commit into
COLAZERO2 wants to merge 1 commit into
