Skip to content

Instantly share code, notes, and snippets.

View KimBue's full-sized avatar

Kim Buergl KimBue

View GitHub Profile
@KimBue
KimBue / summary.txt
Created November 4, 2019 15:18
XLNet Keras summary
Model: "model_1"
__________________________________________________________________________________________________
Layer (type) Output Shape Param # Connected to
==================================================================================================
Input-Token (InputLayer) [(None, 128)] 0
__________________________________________________________________________________________________
Embed-Token (EmbeddingRet) [(None, 128, 768), ( 24576000 Input-Token[0][0]
__________________________________________________________________________________________________
Masking (CreateMask) (None, 128) 0 Input-Token[0][0]
__________________________________________________________________________________________________