Closed ziyunxuanzzz closed 5 months ago
Is 12GB of memory sufficient for training this model with batch-size 1?
Yes. If you're using a larger model and running out of GPU memory, you can try using a smaller image input size
Is 12GB of memory sufficient for training this model with batch-size 1?