Mike Clark and GitHub
87fa9ede4d
ignore parameters with no gradient
...
Is this the right approach?, perhaps it would be better to show trainable vs nontrainable parameters. Or sill use nontrainable parameters to estimate macs
2019-07-07 02:11:36 +00:00
Namhyuk Ahn and GitHub
6a722e9afd
Merge pull request #3 from wassname/patch-1
...
Add example of large resnet model
2019-06-06 13:41:46 +09:00
Mike Clark and GitHub
b5200b41dd
Add example of large resnet model
...
I thought perhaps we could add an example of the outputs on a large model. I noticed that some of these summary packages break down in the presence of large models, even though these are the ones you most need to summarise.
2019-06-06 00:54:02 +00:00
Namhyuk Ahn
d61b3d7032
Update to v1.2.0
2019-06-05 19:17:05 +09:00
Namhyuk Ahn
36b54e837a
Fix bugs
2019-06-05 19:02:56 +09:00
Namhyuk Ahn and GitHub
51b2c50f8b
Merge pull request #2 from wassname/master
...
Use module path names, use pandas
2019-06-05 18:16:19 +09:00
Mike Clark and GitHub
755069ccb3
don't print nans
2019-06-05 05:45:39 +00:00
Mike Clark and GitHub
2db8410f43
fix duplicate keys
2019-06-05 05:39:44 +00:00
Mike Clark and GitHub
24cdc4958c
Show changed outputs
2019-06-05 05:39:12 +00:00
Mike Clark and GitHub
be8002024e
include module path in name, return dataframe
2019-06-05 05:27:28 +00:00
Namhyuk Ahn
558b0ec4e5
Update to v1.1.0
2019-02-03 16:38:49 +09:00
Namhyuk Ahn
7e08acaf7a
Apply lint and use no_grad
2019-02-03 16:36:39 +09:00
Namhyuk Ahn and GitHub
51fdd3c638
Update torchsummaryX.py
2018-08-05 13:03:29 +09:00
Namhyuk Ahn and GitHub
a6ad5f983b
Update README.md
2018-07-30 09:53:27 +09:00
Namhyuk Ahn
e0df04c636
Packaging to v1.0.0
2018-07-30 09:51:51 +09:00
Namhyuk Ahn
786ea000e1
Update README.md
2018-07-29 01:57:24 +09:00
Namhyuk Ahn
40a92a4e28
Merge branch 'master' of https://github.com/nmhkahn/torchsummaryX
2018-07-29 01:56:00 +09:00
Namhyuk Ahn
a0af764402
Fix RNN
2018-07-29 01:55:45 +09:00
Namhyuk Ahn and GitHub
524c9936ef
Update README.md
2018-07-28 15:57:18 +09:00
Namhyuk Ahn
e4cf386bed
Move to torchsummaryX/
2018-07-28 14:38:14 +09:00
Namhyuk Ahn and GitHub
7767887d94
Update README.md
2018-07-28 14:37:11 +09:00
Namhyuk Ahn
049ee409c0
Add torchsummaryX
2018-07-28 14:11:11 +09:00
Namhyuk Ahn
f810921e82
Initial commit
2018-07-28 14:11:01 +09:00