liyiying / Feature_Critic

Feature-Critic Networks for Heterogeneous Domain Generalisation
52 stars 11 forks source link

About the GPU memory increase during training #9

Open QiaoXiao7282 opened 3 years ago

QiaoXiao7282 commented 3 years ago

When I train the model using this code, the GPU memory increases up to out of memory. Do you have any solution about that? Thanks alot, looking forward to your reply.