Skip to content

Commit 897096a

Browse files
committed
Simplified example for v2.1.0
1 parent 47ec22f commit 897096a

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

examples/example.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -99,13 +99,7 @@ def __len__(self):
9999
model.eval()
100100
# enable fused and locality-aware memory access optimization
101101
torchsparse.backends.benchmark = True # type: ignore
102-
# enable adaptive grouping optimization
103-
torchsparse.tune(
104-
model=model,
105-
data_loader=dataflow,
106-
n_samples=10,
107-
collect_fn=lambda data: data['input'],
108-
)
102+
109103
with torch.no_grad():
110104
for k, feed_dict in enumerate(dataflow):
111105
inputs = feed_dict['input'].to(device=args.device).half()

0 commit comments

Comments
 (0)