params-based/gpt2/14/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,
            768
        ],
        [
            1,
            0,
            50257
        ]
    ],
    "flops": [
        0,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        501815808
    ],
    "times": [
        8.749961853027344e-05,
        0.0017244815826416016,
        0.002473115921020508,
        0.009609460830688477,
        0.0019423961639404297,
        0.004061698913574219,
        0.002203702926635742,
        0.002424001693725586,
        0.002797365188598633,
        0.0033464431762695312,
        0.016410112380981445,
        0.0026390552520751953,
        0.006195783615112305,
        0.004674434661865234
    ],
    "params": [
        39383808,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        38598912
    ]
}