From 8dc64e0e21a7bd6e3f5b76bea6e63ba8fccfb41b Mon Sep 17 00:00:00 2001 From: Cedric Scheerlinck Date: Tue, 29 Sep 2020 18:03:55 +1000 Subject: [PATCH] remove random break --- inference.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/inference.py b/inference.py index 98912de..c76eb52 100644 --- a/inference.py +++ b/inference.py @@ -97,8 +97,6 @@ def main(args, model): model.reset_states() for i, item in enumerate(tqdm(data_loader)): - if i > 1000: - break voxel = item['events'].to(device) if not args.color: voxel = crop.pad(voxel)