update app

app.py

@@ -26,7 +26,7 @@ def get_num_parameters(model):
 
 # %% ../nbs/00_benchmark.ipynb 11
 @torch.inference_mode()
-def evaluate_cpu_speed(model, dummy_input, warmup_rounds=50, test_rounds=100):
+def evaluate_cpu_speed(model, dummy_input, warmup_rounds=5, test_rounds=20):
     device = torch.device("cpu")
     model.eval()
     model.to(device)
@@ -61,7 +61,7 @@ def get_model_macs(model, inputs) -> int:
 
 # %% ../nbs/00_benchmark.ipynb 16
 @torch.inference_mode()
-def evaluate_emissions(model, dummy_input, warmup_rounds=50, test_rounds=100):
+def evaluate_emissions(model, dummy_input, warmup_rounds=5, test_rounds=20):
     device = torch.device("cpu")
     model.eval()
     model.to(device)