Skip to content

Pull requests: sksq96/pytorch-summary

New pull request New
32 Open 24 Closed
32 Open 24 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

feat: allow nested list/tuple input/output and None type
#200 opened Jan 28, 2024 by cijinsama Loading…
4
Fix None Type error while using MultiHeadAttention
#191 opened Oct 26, 2022 by GCS-ZHN Loading…
feat: allow more input shapes and weight locations
#189 opened Aug 5, 2022 by ClashLuke Loading…
Add license to setup.py
#187 opened Jun 9, 2022 by normanrz Loading…
Added OrderedDict as an output datatype
#184 opened Mar 19, 2022 by SoumyadeepB Loading…
Return model size info
#182 opened Mar 14, 2022 by mhamdan91 Loading…
support for list as output item, not only tensors
#167 opened Mar 11, 2021 by leora-betesh Loading…
Fix handling input_size with multi-input
#166 opened Mar 11, 2021 by ahmedhshahin Loading…
3
Fix the multi-output, dict-input, parameter counting and calculation overflow problem.
#165 opened Feb 27, 2021 by cainmagi Loading…
Fix parameter count
#142 opened Aug 14, 2020 by luisherrmann Loading…
Minor tweaks and quick fix
#119 opened Mar 9, 2020 by leriomaggio Loading…
1
feat: Default device is set to model device
#117 opened Mar 5, 2020 by frgfm Loading…
3
fix: Fixed hook for DenseNet
#116 opened Mar 5, 2020 by frgfm Loading…
11
Use GPU only when available else default to CPU
#114 opened Feb 26, 2020 by Gokkulnath Loading…
1
add checks for atomic values in output
#111 opened Feb 22, 2020 by wyc1997 Loading…
Added two new parameters to summary: verbose and csv. Verbose shows m…
#108 opened Jan 10, 2020 by eplatero97 Loading…
Allow user to specify dtype
#104 opened Dec 23, 2019 by tberckmann Loading…
Add dictionary support
#101 opened Nov 20, 2019 by Morpheus3000 Loading…
1
Multi-gpu compatibility: torch-summary for 'cuda:1' and so on
#93 opened Sep 21, 2019 by adithyavis Loading…
1
torchsummary() extensions by input_initializer, dtype
#89 opened Aug 12, 2019 by DrStoop Loading…
count running_mean and running_var params for BN
#88 opened Aug 9, 2019 by mostafaelhoushi Loading…
2
Add a function that can control the depth of output and display the structure of the model
#83 opened Jul 19, 2019 by sangyx Loading…
New functionality: use torchsummary to build pytorch model with scalable input shape
#80 opened Jul 4, 2019 by ncble Loading…
Fixed params count and input size for multiple inputs
#71 opened Jun 1, 2019 by ajaynarayanan Loading…
1
Accomodation for various datatypes
#70 opened May 28, 2019 by kr-hari Loading…
ProTip! Follow long discussions with comments:>50.