CUDA error: out of memory when load models

Check this thread: Out of memory error when resume training even though my GPU is empty

You need to load onto CPU first. Someone should file a bug report about this…

1 Like