flops-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": [
        92112384,
        92112384,
        92112384,
        184224768,
        184224768,
        184224768,
        184224768,
        92162304,
        501765888
    ],
    "times": [
        0.0027399063110351562,
        0.003090381622314453,
        0.002351999282836914,
        0.004258155822753906,
        0.004597663879394531,
        0.005335092544555664,
        0.004565238952636719,
        0.004034996032714844,
        0.005158901214599609
    ],
    "params": [
        46471680,
        7087872,
        7087872,
        14175744,
        14175744,
        14175744,
        14175744,
        7089408,
        38597376
    ]
}