training with one GPU

[wo0423]

excuse me,i want to know how to train with only one GPU, can you help me? thanks a lot.

[charlesCXK]

Hi, we highly recommend using 4 or 8 GPUs to train the model, since the performance is very sensitive to the batch size.

To train with one GPU, you could set a smaller batch size and linearly reduce the learning rate.

[wo0423]

got it, due to limited conditions, I only have one GPU at my disposal, very appreciate for your patient reply!

[wo0423]

Hello,i’m sorry to bother you again,there is a problem that has puzzled me for a long time. I really can't figure out where to correct it.I trained with one gpu. I made the following modifications to the code(train.py):

and

and

and

The problem is that my training results are poor.I don't know what the problem is.

[XieZhiyao21]

do you solve this problem?i want to know how to deal with that too.