Zekun Wu commited on
Commit
0c50b28
1 Parent(s): 9f3ad0b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def calculate_flops_hf(model_name, input_shape, access_token, bp_factor):
43
  bp_factor=bp_factor)#, output_unit=output_unit)
44
  print(f"Data: {data}")
45
 
46
- total_flops = data[0]['Forward+Backward FLOPs'] * 10 ** 9 # Convert GFLOPs to FLOPs
47
  threshold_flops = 10 ** 25
48
  meets_criteria = total_flops > threshold_flops
49
 
 
43
  bp_factor=bp_factor)#, output_unit=output_unit)
44
  print(f"Data: {data}")
45
 
46
+ total_flops = data[0]['Forward+Backward FLOPs']
47
  threshold_flops = 10 ** 25
48
  meets_criteria = total_flops > threshold_flops
49