GirishaBuilds01 commited on
Commit
f2d4330
·
verified ·
1 Parent(s): 68aeded

Create utils/energy.py

Browse files
Files changed (1) hide show
  1. utils/energy.py +8 -0
utils/energy.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ def energy_cost(precision, hardware):
2
+ base = {"CPU": 1.0, "GPU": 0.7, "EDGE": 0.5}[hardware]
3
+ return {
4
+ "FP32": 1.0,
5
+ "FP16": 0.6,
6
+ "INT8": 0.35,
7
+ "INT4": 0.2
8
+ }[precision] * base