Created by: yeyupiaoling

我使用的是docker进行训练，显存为11G，在训练过程中会出现不断吃显存，直到显存不足挂掉。设置小batch_size也没有用的。