dd794a6ab3
add tests
2019-11-19 15:23:49 +08:00
006b7e06c4
let it work with modules with are layers
2019-11-19 15:23:40 +08:00
1915788513
Update README.md
2019-07-07 13:40:02 +09:00
bca7d973dd
Fix #6 and apply lint
2019-07-07 13:34:51 +09:00
93123d8798
Merge pull request #7 from wassname/patch-2
...
ignore parameters with no gradient
2019-07-07 12:54:37 +09:00
f8896c6250
max rows to 600
2019-07-07 03:38:56 +00:00
a903488e39
show non trainable params in summary, use pandas to display engineering units
...
This shows a trainable parameter in the full data frame and trainable vs non-trainable params in the summary. This should be like keras [example](https://github.com/keras-team/keras/pull/11795#issue-235955201 ).
I also changed the pandas display so that instead of putting (M) or (K) manually in the label, pandas appends it to the number automatically.
2019-07-07 03:35:52 +00:00
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
6a722e9afd
Merge pull request #3 from wassname/patch-1
...
Add example of large resnet model
2019-06-06 13:41:46 +09:00
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
d61b3d7032
Update to v1.2.0
2019-06-05 19:17:05 +09:00
36b54e837a
Fix bugs
2019-06-05 19:02:56 +09:00
51b2c50f8b
Merge pull request #2 from wassname/master
...
Use module path names, use pandas
2019-06-05 18:16:19 +09:00
755069ccb3
don't print nans
2019-06-05 05:45:39 +00:00
2db8410f43
fix duplicate keys
2019-06-05 05:39:44 +00:00
24cdc4958c
Show changed outputs
2019-06-05 05:39:12 +00:00
be8002024e
include module path in name, return dataframe
2019-06-05 05:27:28 +00:00
558b0ec4e5
Update to v1.1.0
2019-02-03 16:38:49 +09:00
7e08acaf7a
Apply lint and use no_grad
2019-02-03 16:36:39 +09:00
51fdd3c638
Update torchsummaryX.py
2018-08-05 13:03:29 +09:00
a6ad5f983b
Update README.md
2018-07-30 09:53:27 +09:00
e0df04c636
Packaging to v1.0.0
2018-07-30 09:51:51 +09:00
786ea000e1
Update README.md
2018-07-29 01:57:24 +09:00
40a92a4e28
Merge branch 'master' of https://github.com/nmhkahn/torchsummaryX
2018-07-29 01:56:00 +09:00
a0af764402
Fix RNN
2018-07-29 01:55:45 +09:00
524c9936ef
Update README.md
2018-07-28 15:57:18 +09:00
e4cf386bed
Move to torchsummaryX/
2018-07-28 14:38:14 +09:00
7767887d94
Update README.md
2018-07-28 14:37:11 +09:00
049ee409c0
Add torchsummaryX
2018-07-28 14:11:11 +09:00
f810921e82
Initial commit
2018-07-28 14:11:01 +09:00
wassname