跑了几个epoch之后，显示显存不足，
报错finished with exit code -1073740791 (0xC0000409)
那就把batch_size设置的小一些。比如先改成现有的一半。
我的显存是2g的，batch_size都改成2了还是不行，就跑的SSD例程........