Program doesn't exit after training finished

[umdreamer]

After reach the end of line in file src/train.py. The program still run, but the CPU and GPU usages are both 0%. The memory was not released.

[buptlmk]

Do you have reproduce the result?

[umdreamer]

I think this is because CUDA doesn't release memory for the program. Just use Ctrl+C to break it down. Everything seems OK except that it can not exit automatically.

I have't reproduced the full result. I'm still running the experiments.

[buptlmk]

ok，Thank you. Please contact me if you have reproduced the result, we can help each other.