test.py

[jfprieur]

Getting the following error message when running the command

export PYTHONPATH=/workspace
CUDA_VISIBLE_DEVICES=0 python tools/test.py
configs/oneformer3d_qs_radius16_qp300_2many.py
work_dirs/clean_forestformer/epoch_3000_fix.pth

/opt/conda/lib/python3.10/site-packages/MinkowskiEngine/init.py:36: UserWarning: The environment variable OMP_NUM_THREADS not set. MinkowskiEngine will automatically set OMP_NUM_THREADS=16. If you want to set OMP_NUM_THREADS manually, please export it on the command line before running a python script. e.g. export OMP_NUM_THREADS=12; python your_program.py. It is recommended to set it below 24.
warnings.warn(
/opt/conda/lib/python3.10/site-packages/mmdet3d/evaluation/functional/kitti_utils/eval.py:10: NumbaDeprecationWarning: The 'nopython' keyword argument was not supplied to the 'numba.jit' decorator. The implicit default value for this argument is currently False, but it will be changed to True in Numba 0.59.0. See https://numba.readthedocs.io/en/stable/reference/deprecation.html#deprecation-of-object-mode-fall-back-behaviour-when-using-jit for details.
def get_thresholds(scores: np.ndarray, num_gt, num_sample_pts=41):
Applying spconv checkpoint fix in-memory...
Traceback (most recent call last):
File "/workspace/tools/test.py", line 175, in
main()
File "/workspace/tools/test.py", line 123, in main
checkpoint = torch.load(args.checkpoint, map_location='cpu')
NameError: name 'torch' is not defined

Adding import torch to test.py solved the error.