Closed BuyMyMojo closed 3 years ago
Hello! Your validation images are probably too big, larger than what can be handled with the available VRAM.
They can be few and small and only required for you to visualize how the model is doing, it doesn't change the end results.
Alright cool thank you!
It uses about 3GB when training and then about 5.2 when it starts the validation then it crashes. My training data is 512x512 jpg files, one frame: https://imgur.com/a/MulGiTE
GPU: 2070 super CPU: 5600x cuda 11 installed torch==1.9.1+cu111 torchvision==0.10.1+cu111
Complete powershell output: