Add memory demanding to pytorch models

bob/bio/face/config/baseline/afffe.py

@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import afffe_baseline
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return afffe_baseline(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return afffe_baseline(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)
 

bob/bio/face/config/baseline/iresnet100.py

@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet100
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return iresnet100(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return iresnet100(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)
 

bob/bio/face/config/baseline/iresnet34.py

@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet34
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return iresnet34(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return iresnet34(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)
 

bob/bio/face/config/baseline/iresnet50.py

@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet50
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return iresnet50(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return iresnet50(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)