search

AlexHex7 / DRML_pytorch

Implementation of Deep Region and Multi-Label Learning for Facial Action Unit Detection.
45 stars 14 forks source link

Need Help QAQ #1

Open xjrelc opened 5 years ago

xjrelc commented 5 years ago

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 commented 5 years ago

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