Thank you for your answer Frank .it helps!  btw do you have any experience on google colab cuda out og memory issues. I try to train around 150,000 images (7kb each)  , it seems I dunt have enough resources on GPU , but train using cpu really really too slow .