Hi, I see you said the train at least needed 4 a100. I wonder whether 2 a100 is enough when i set the batch_size 1 or set other something smaller?