Skip to content
Snippets Groups Projects

Resolve "ArcFace (MXNET) + IJBC lead to memory error"

Merged Laurent COLBOIS requested to merge mxnet-demanding into master
2 files
+ 17
6
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -9,6 +9,9 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
if "database" in locals():
annotation_type = database.annotation_type
fixed_positions = database.fixed_positions
memory_demanding = (
database.memory_demanding if hasattr(database, "memory_demanding") else False
)
else:
annotation_type = None
fixed_positions = None
@@ -16,7 +19,7 @@ else:
def load(annotation_type, fixed_positions=None):
transformer = embedding_transformer_112x112(
ArcFaceInsightFace(), annotation_type, fixed_positions, color_channel="rgb"
ArcFaceInsightFace(memory_demanding=memory_demanding), annotation_type, fixed_positions, color_channel="rgb"
)
algorithm = Distance()
Loading