Test error

[arsalan311]

I am getting this error when I run the test file Traceback (most recent call last): File "tools/test.py", line 211, in main() File "tools/test.py", line 207, in main eval_single_ckpt(model, test_loader, args, eval_output_dir, logger, epoch_id, dist_test=dist_test) File "tools/test.py", line 75, in eval_single_ckpt result_dir=eval_output_dir, save_to_file=args.save_to_file File "/home/arsalan/CLOC_Project/LidarMTL/LidarMTL/tools/eval_utils/eval_utils.py", line 58, in eval_one_epoch pred_dicts, ret_dict = model(batch_dict) File "/home/arsalan/CLOC_Project/second/lib/python3.6/site-packages/torch/nn/modules/module.py", line 532, in call result = self.forward(*input, **kwargs) File "/home/arsalan/CLOC_Project/LidarMTL/LidarMTL/pcdet/models/detectors/LidarMTL.py", line 21, in forward pred_dicts, recall_dicts = self.post_processing(batch_dict) File "/home/arsalan/CLOC_Project/LidarMTL/LidarMTL/pcdet/models/detectors/detector3d_template.py", line 190, in post_processing V.visualize_pts(batch_dict) File "/home/arsalan/CLOC_Project/LidarMTL/LidarMTL/tools/visual_utils/visualize_utils.py", line 75, in visualize_pts pts = pts.cpu().numpy() AttributeError: 'dict' object has no attribute 'cpu'

also In detector_3D.py V.visualize_voxel_prediction(batch_dict) is called but this function is not present in the file. I also verified it with PCDET