Need Help QAQ

[xjrelc]

Really Sorry but just want to ask that how to use checkpoints for this model? as i try to run the model with disfa or other bigger dataset it will came out the error about "CUDA run out of memory" ,QAQ, i seach the way to solve it but i have no idea to use the supported way of checkpoints in this model.
Any ways, thanks!!

[AlexHex7]

@xjrelc Hi. Maybe the img size is too larger? or you can try to decrease the batch size.