src/benchmark_metrics.py [122:128]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            "resnet101": {  # model name
                "ml.g5.8xlarge": {1: {224: 2500}}  # instance  # Num nodes  # batch size : threshold
            },
            "GPT-2": {  # model name
                "ml.g5.8xlarge": {1: {75: 2000}}  # instance  # Num nodes  # batch size : threshold
            },
        },
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/benchmark_metrics.py [130:136]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            "resnet101": {  # model name
                "ml.g5.8xlarge": {1: {224: 2500}}  # instance  # Num nodes  # batch size : threshold
            },
            "GPT-2": {  # model name
                "ml.g5.8xlarge": {1: {75: 2000}}  # instance  # Num nodes  # batch size : threshold
            },
        },
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -