diff --git a/README.md b/README.md index 636bd9b..0d63695 100644 --- a/README.md +++ b/README.md @@ -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 ---------------------------------------------------------------------------------------------------- ```