fjw

Hi, It could be that the pytorch checkpointing function is not supporting apex. Did you try torch.cuda.amp?

Our implementation of auto parsing graph is depending on torch.jit and quite volatile with pytorch version. If you have manual parse_graph function it can definitely work with 1.6. For auto...

Hi, Autoparse should support all the networks in benchmark.py. Can you remove the "try" at https://github.com/lordfjw/OptimalGradCheckpointing/blob/main/benchmark.py#L167 and paste the error message? Can you share the pytorch version? It might also...