search

zhijian-liu / torchprofile

A general and accurate MACs / FLOPs profiler for PyTorch models
https://pypi.org/project/torchprofile/
MIT License
560 stars 38 forks source link

[major] fix nn.ConvTransposeNd #2

Closed lmxyy closed 4 years ago