Questions about "save_debug_2d_images" function

[912267428]

My GPU is 3090, so I cannot use the Torch1.4.0+cuda10.1 environment mentioned by the author. The environment I am using is Torch1.11.0+cuda11.3, which is the lowest version of the 30 series. I set gpus to "0" in the configuration file, num_ image is set to 2K, and there is a problem with the image shown in the above figure during operation. This problem is caused by the save_debug_2d_images function. The error prompt indicates that the location of the error is in the library.

I hope someone can help me solve this problem. Thank you very much

[AlvinYH]

Thanks for your interest in our work. We've modified the code and solved the aforementioned problems. Please pull the recent release. Now it can support a higher PyTorch version and the visualization function should work fine.