search

Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.
MIT License
4.9k stars 528 forks source link

adding hook function for nn.Transformer #140

Closed HaoKang-Timmy closed 3 years ago

HaoKang-Timmy commented 3 years ago

not available for custom_encoder and custom_decoder