wassname/torchsummaryX
1
0
Fork 0
mirror of https://github.com/wassname/torchsummaryX.git synced 2026-06-27 17:50:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Namhyuk Ahn
2018-07-29 01:57:24 +09:00
parent 40a92a4e28
commit 786ea000e1
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4 -4
View File
@@ -75,16 +75,16 @@ summary(Net(), inputs)
----------------------------------------------------------------------------------------------------
Layer Kernel Shape Output Shape # Params (K) # Mult-Adds (M)
====================================================================================================
0_Embedding [300, 20] [100, 1, 300] 6.00 1.80
1_LSTM - [100, 1, 512] 3,768.32 376.01
0_Embedding [300, 20] [100, 1, 300] 6.00 0.01
1_LSTM - [100, 1, 512] 3,768.32 3.76
weight_ih_l0 [2048, 300]
weight_hh_l0 [2048, 512]
weight_ih_l1 [2048, 512]
weight_hh_l1 [2048, 512]
2_Linear [512, 20] [100, 1, 20] 10.26 0.20
2_Linear [512, 20] [100, 1, 20] 10.26 0.01
====================================================================================================
# Params: 3,784.58K
# Mult-Adds: 378.02M
# Mult-Adds: 3.78M
----------------------------------------------------------------------------------------------------
```