search

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

Pass version to _parse_node_inputs recursive call #8

Closed senysenyseny16 closed 3 years ago

1adrianb commented 3 years ago

Thanks a lot for spotting and fixing this!