Commit Graph

23 Commits

Author SHA1 Message Date
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