KeyError: 'occ'

[YANG-Deep]

Thank you for your outstanding work! I have some questions about the visualization of occupancy map.
I saved result[0]['occ'] into result.pkl during evaluation, and visualized the map with render_cfg.with_occ_map=True.
However, when I run the code, I find:

1. if 'topk_query_ins_segs' in outputs[k]['occ']:
    KeyError: 'occ'
   

2. Also when I search something named ['topk_query_ins_segs'], I found this code only occurred in /visualize/run.py
   so I can not visualize the occupancy map correctly.

I would appreciate it if you could tell me what is wrong with my operation and how to visualize the occformer.

[Ep11phany]

Hello there, I found the same problem, and I was wondering if you have found the reason for and solution to visualizing the occ map?