Get your AI tool to write its output to a long on disk. Get another program to log ram and vram usage periodically as well. As my top suspicion is your application has a memory leak, like many such applications do, and it just gets the system to hang when it runs too long. There are ways to fix this too, like PyTorch allocation variables, inserting some manual garbage collection, or just periodically restarting it.