Also fixed a potential RuntimeError "RuntimeError: unable to open shared memory object" when many training sessions are running.