utils/energy.py

def energy_cost(precision, hardware):
    base = {"CPU": 1.0, "GPU": 0.7, "EDGE": 0.5}[hardware]
    return {
        "FP32": 1.0,
        "FP16": 0.6,
        "INT8": 0.35,
        "INT4": 0.2
    }[precision] * base