params-based/gpt2/12/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,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            50257
        ]
    ],
    "flops": [
        0,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        276337152,
        501815808
    ],
    "times": [
        0.00015115737915039062,
        0.0025413036346435547,
        0.012193441390991211,
        0.015429019927978516,
        0.0021784305572509766,
        0.0029501914978027344,
        0.002393960952758789,
        0.002190828323364258,
        0.002512693405151367,
        0.002668619155883789,
        0.012943267822265625,
        0.004745006561279297
    ],
    "params": [
        39383808,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        21263616,
        38598912
    ]
}