yyang181/NTIRE23-VIDEO-COLORIZATION

Slow Processing 1,5FPS

Opened this issue · 0 comments

The processing speed is quite slow. I tried to implement orch.utils.data.DataLoader and unfortunately I couldn't get it to work. The VRAM is also too busy. This can be supplemented with torch.cuda.empty_cache() for the memory-hungry loops. If someone has solved the performance problem, I would be happy, thank you