Spaces:
Runtime error
Runtime error
Zekun Wu
commited on
Commit
·
f4c36a8
1
Parent(s):
8e01763
add
Browse files
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
|