memory cost

[VLadImirluren]

I haven't as large compute resource as yours. This make me hard to reproduce your work. Can you share some trick of reduce the memory cost but reach the same effect? (In brief, how to use lower batchsize but same result)