flops-based/gpt2/5/info.json

{
    "shapes": [
        [
            1,
            13
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            50257
        ]
    ],
    "flops": [
        276337152,
        276337152,
        276337152,
        276387072,
        501765888
    ],
    "times": [
        0.005208730697631836,
        0.005757808685302734,
        0.00614166259765625,
        0.005856513977050781,
        0.004487514495849609
    ],
    "params": [
        60647424,
        21263616,
        21263616,
        21265152,
        38597376
    ]
}