PReLU and ELU was not add to the list into isinstance - Githubissues

Swall0w / torchstat

Model analyzer in PyTorch

MIT License

1.47k stars 142 forks source link

PReLU and ELU was not add to the list into isinstance #2

Closed gdshen closed 6 years ago

gdshen commented 6 years ago

PReLU and ELU was not add to the list into isinstance https://github.com/Swall0w/torchstat/blob/621549e458971e12eb70a118fc8a7d9b0247abfd/torchstat/compute_flops.py#L61

although they are added in https://github.com/Swall0w/torchstat/blob/621549e458971e12eb70a118fc8a7d9b0247abfd/torchstat/compute_flops.py#L13

Swall0w commented 6 years ago

Thank you for the information. I've fixed it :thumbsup: