Update test.py

MatteoMendula · Apr 28, 2024 · 53b59d1 · 53b59d1
1 parent 3004a11
commit 53b59d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -42,7 +42,7 @@ def test_sample(test_loader):
         times[i] = starter.elapsed_time(ender)
 
         metric.compute(pred, ele_gt, ele_mask)
-        with open('./bev_pred/' + cur_time + '.pkl', 'wb') as f:
+        with open('./bev_pred/' + cur_time[0] + '.pkl', 'wb') as f:
             pickle.dump(pred.squeeze().data.cpu(), f)
 
     mean_time = times.mean().item()