Zekun Wu commited on
Commit
f4c36a8
·
1 Parent(s): 8e01763
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -41,6 +41,8 @@ def calculate_flops_hf(model_name, input_shape, access_token, bp_factor, output_
41
  data, return_print = calculate_flops_in_hugging_space(model_name=model_name, empty_model=model,
42
  access_token=access_token, input_shape=input_shape,
43
  bp_factor=bp_factor, output_unit=output_unit)
 
 
44
  df = data['value']
45
  total_flops = df['Forward+Backward FLOPs'][0] * 10 ** 9 # Convert GFLOPs to FLOPs
46
  threshold_flops = 10 ** 25
 
41
  data, return_print = calculate_flops_in_hugging_space(model_name=model_name, empty_model=model,
42
  access_token=access_token, input_shape=input_shape,
43
  bp_factor=bp_factor, output_unit=output_unit)
44
+ print(f"Data: {data}")
45
+
46
  df = data['value']
47
  total_flops = df['Forward+Backward FLOPs'][0] * 10 ** 9 # Convert GFLOPs to FLOPs
48
  threshold_flops = 10 ** 25