params-based/gpt2/9/info.json

{
    "shapes": [
        [
            1,
            0
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            50257
        ]
    ],
    "flops": [
        0,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        184224768,
        460561920,
        501815808
    ],
    "times": [
        0.0003204345703125,
        0.04195833206176758,
        0.05009007453918457,
        0.02313685417175293,
        0.00899505615234375,
        0.005730152130126953,
        0.02529168128967285,
        0.008275747299194336,
        0.004660844802856445
    ],
    "params": [
        39383808,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        14175744,
        35439360,
        38598912
    ]
}