Apr 17, 2023
No, but I did commented out all the validation and save check-pointing code snippet before I do the training. I am able to train with batch size of two on a 6GB GPU on 512x512 images.
Written by Ng Wai Foong
Senior AI Engineer@Yoozoo | Content Writer #NLP #datascience #programming #machinelearning | Linkedin: https://www.linkedin.com/in/wai-foong-ng-694619185/