Skip to content
Snippets Groups Projects

`memory_demanding` for TF based transformers

Merged Tiago de Freitas Pereira requested to merge memory_demanding into master
8 files
+ 133
40
Compare changes
  • Side-by-side
  • Inline
Files
8
@@ -7,10 +7,14 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
@@ -7,10 +7,14 @@ from bob.bio.base.pipelines.vanilla_biometrics import (
VanillaBiometricsPipeline,
VanillaBiometricsPipeline,
)
)
memory_demanding = False
if "database" in locals():
if "database" in locals():
annotation_type = database.annotation_type
annotation_type = database.annotation_type
fixed_positions = database.fixed_positions
fixed_positions = database.fixed_positions
 
memory_demanding = (
 
database.memory_demanding if hasattr(database, "memory_demanding") else False
 
)
 
else:
else:
annotation_type = None
annotation_type = None
fixed_positions = None
fixed_positions = None
@@ -19,7 +23,9 @@ else:
@@ -19,7 +23,9 @@ else:
def load(annotation_type, fixed_positions=None):
def load(annotation_type, fixed_positions=None):
transformer = embedding_transformer_160x160(
transformer = embedding_transformer_160x160(
FaceNetSanderberg_20170512_110547(), annotation_type, fixed_positions
FaceNetSanderberg_20170512_110547(memory_demanding=memory_demanding),
 
annotation_type,
 
fixed_positions,
)
)
algorithm = Distance()
algorithm = Distance()
Loading