search

frgfm / torch-scan

Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
https://frgfm.github.io/torch-scan/latest
Apache License 2.0
208 stars 22 forks source link

feat: Added FLOPS estimation support for nn.Transformer #48

Closed frgfm closed 3 years ago

frgfm commented 3 years ago

Following up on #45, this PR introduces the following modifications:

codecov[bot] commented 3 years ago

Codecov Report

Merging #48 (7c03cc1) into master (de4472e) will increase coverage by 0.42%. The diff coverage is 97.32%.

Impacted file tree graph

@@            Coverage Diff             @@
##           master      #48      +/-   ##
==========================================
+ Coverage   89.93%   90.35%   +0.42%     
==========================================
  Files          10       10              
  Lines         616      674      +58     
==========================================
+ Hits          554      609      +55     
- Misses         62       65       +3
Impacted Files Coverage Δ
torchscan/modules/flops.py 97.57% <97.19%> (-1.49%) :arrow_down:
torchscan/crawler.py 82.08% <100.00%> (ø)