flops-based/gpt2/5/info.json

{
    "shapes": [
        [
            1,
            0
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            768
        ],
        [
            1,
            0,
            50257
        ]
    ],
    "flops": [
        276337152,
        276337152,
        276337152,
        276387072,
        501765888
    ],
    "times": [
        0.006539821624755859,
        0.005666017532348633,
        0.006500244140625,
        0.005806684494018555,
        0.005162477493286133
    ],
    "params": [
        60647424,
        21263616,
        21263616,
        21265152,
        38597376
    ]
}