Resolve "ArcFace (MXNET) + IJBC lead to memory error"
Closes #42 (closed)
Edited by Laurent COLBOIS
Merge request reports
Activity
added 1 commit
- 71387a13 - Enable memory_demanding mode (samplewise processing) for MXNet models
This enables to run MXNet models in memory_demanding mode (meaning samplewise forward passes instead of batchwise). I followed the same structure as in TF2 models. The memory_demanding parameter can be recovered from the input database. With this fix, I am able to run the IJBC evaluation with ArcFace.
I also made sure to get the exact same scores on Mobio before and after this change, and the exact same scores on Mobio running with and without the memory_demanding option.
ping @tiago.pereira
awesome @lcolbois :-)
It looks perfect. I'll let you merge.
Do you mind sharing a ROC plot of your experiment?
Thanks
assigned to @lcolbois
mentioned in commit cb469550
Please register or sign in to reply