Thanks for reply me back.
Currently, i have following configuration:
Train Images: 1800 - Valid: 450
batch_size = 12
num_workers = 0
And my average GPU temperature remain 80 °C, it is normal? or here, i can also try something else?