RuntimeError: step!=1 is currently not supported

1adrianb / pytorch-estimate-flops

Estimate/count FLOPS for a given neural network using pytorch

https://www.adrianbulat.com

BSD 3-Clause "New" or "Revised" License

305 stars 22 forks source link

RuntimeError: step!=1 is currently not supported #13

Open ResonWang opened 2 years ago

ResonWang commented 2 years ago

Unable to import torch.fx, you pytorch version may be too old. FX is unsupported on your pytorch version, falling back to JIT RuntimeError: step!=1 is currently not supported