Simplified example for v2.1.0

mit-han-lab · Jul 1, 2023 · 897096a · 897096a
1 parent 47ec22f
commit 897096a
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/examples/example.py b/examples/example.py
@@ -99,13 +99,7 @@ def __len__(self):
     model.eval()
     # enable fused and locality-aware memory access optimization
     torchsparse.backends.benchmark = True  # type: ignore
-    # enable adaptive grouping optimization
-    torchsparse.tune(
-        model=model,
-        data_loader=dataflow,
-        n_samples=10,
-        collect_fn=lambda data: data['input'],
-    )
+
     with torch.no_grad():
         for k, feed_dict in enumerate(dataflow):
             inputs = feed_dict['input'].to(device=args.device).half()