dust-removal
dust-removal

> Hi @dust-removal , what is your cuda version? In my system is 12.2. In the conda environment is 10.2 
>   Hi, I also want to train my own dataset, can you tell me what changes you made to train your own dataset