Add memory demanding to pytorch models

parent 33d75dc8
Pipeline #51490 passed with stage
in 35 minutes and 7 seconds
......@@ -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)
......@@ -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)
......@@ -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)
......@@ -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)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment