How to auto parsing?

[feifeibear]

I run the benchmark.pu with the following warnings.
python benchmark.py --arch resnet18 --device cuda:0
Parsing Computation Graph with torch.jit failed, revert to manual parse_graph function

[jianweif]

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 be pytorch version too old

Thanks