From 8a082e38b5efa4c7a03145d9e69b13e920861b12 Mon Sep 17 00:00:00 2001 From: Jiang anqing Date: Wed, 15 Jun 2022 10:12:46 +0800 Subject: [PATCH] [fix] onnx export bug --- tool/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tool/exporter.py b/tool/exporter.py index 9dbe6a5..c2b52b6 100644 --- a/tool/exporter.py +++ b/tool/exporter.py @@ -121,7 +121,7 @@ def main(): dummy_input['voxels'] = dummy_voxels dummy_input['voxel_num_points'] = dummy_voxel_num dummy_input['voxel_coords'] = dummy_voxel_idxs - dummy_input['batch_size'] = 1 + dummy_input['batch_size'] = torch.tensor(1) torch.onnx.export(model, # model being run dummy_input, # model input (or a tuple for multiple inputs)