yule-li / CosFace

Tensorflow implementation for paper CosFace: Large Margin Cosine Loss for Deep Face Recognition
286 stars 98 forks source link

OOM error #23

Open osman-goni-cse opened 9 months ago

osman-goni-cse commented 9 months ago

Allocator (GPU_0_bfc) ran out of memory trying to allocate 12.0 kiB (rounded to 12288) as requested by op tower_0/l2_normalize_1.

I tried to reduce batch_size, max_nrof_epochs, images_per_person, etc. Nothing works. Can anyone help me?