Mike Clark
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
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
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
51b2c50f8b
Merge pull request #2 from wassname/master
...
Use module path names, use pandas
2019-06-05 18:16:19 +09:00
Mike Clark
755069ccb3
don't print nans
2019-06-05 05:45:39 +00:00
Mike Clark
2db8410f43
fix duplicate keys
2019-06-05 05:39:44 +00:00
Mike Clark
24cdc4958c
Show changed outputs
2019-06-05 05:39:12 +00:00
Mike Clark
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
51fdd3c638
Update torchsummaryX.py
2018-08-05 13:03:29 +09:00
Namhyuk Ahn
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
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
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