params-based/gpt2/13/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,
            768
        ],
        [
            1,
            0,
            50257
        ]
    ],
    "flops": [
        0,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        184224768,
        501815808
    ],
    "times": [
        0.0001957416534423828,
        0.0019838809967041016,
        0.021943330764770508,
        0.0018630027770996094,
        0.002742290496826172,
        0.002980470657348633,
        0.03744912147521973,
        0.0021698474884033203,
        0.0020749568939208984,
        0.006228208541870117,
        0.0027358531951904297,
        0.0039522647857666016,
        0.0047397613525390625
    ],
    "params": [
        39383808,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        14175744,
        38598912
    ]
}