6D pose estimation inference error.

[toshi2k2]

Command run:

CUDA_VISIBLE_DEVICES=0 python3 scripts/inference.py \
    --cate aeroplane \
    --config config/pose_estimation_6d_faster_rcnn.yaml \
    --save_dir exp_6d/pose_estimation_6d_nemo_aeroplane \
    --checkpoint exp_6d/pose_estimation_6d_nemo_aeroplane/ckpts/model_30.pth

Error:

Traceback (most recent call last):
  File "/mnt/NeMo/scripts/inference.py", line 100, in <module>
    main()
  File "/mnt/NeMo/scripts/inference.py", line 96, in main
    inference(cfg)
  File "/mnt/NeMo/scripts/inference.py", line 66, in inference
    results = helper_func_by_task[cfg.task](
KeyError: '6d_pose_estimation'