flops-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": [
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        92112384,
        49920,
        501765888
    ],
    "times": [
        0.009018182754516602,
        0.0032470226287841797,
        0.002666950225830078,
        0.004340648651123047,
        0.004440784454345703,
        0.019390344619750977,
        0.0040683746337890625,
        0.0021238327026367188,
        0.002913236618041992,
        0.002443552017211914,
        0.002220630645751953,
        0.0018863677978515625,
        7.62939453125e-05,
        0.0049130916595458984
    ],
    "params": [
        46471680,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        7087872,
        1536,
        38597376
    ]
}