HWalkingMan

Thanks for your scripts~

I can only run 1k iterations per 12min on the 3090, so I need at least 160h to train the full 800k iterations and 23.9G of GPU memory. Can you...