OptimalGradCheckpointing icon indicating copy to clipboard operation
OptimalGradCheckpointing copied to clipboard

Published 20 hours ago • verified publisher icon lordfjw

How to auto parsing?

Open feifeibear opened this issue 2 years ago • 1 comments

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

feifeibear avatar Nov 02 '21 09:11 feifeibear

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

jianweif avatar Nov 02 '21 19:11 jianweif