Commit 4e2e9722 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

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 ...@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import afffe_baseline
from bob.bio.face.utils import lookup_config_from_database 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") locals().get("database")
) )
def load(annotation_type, fixed_positions=None): def load(annotation_type, fixed_positions=None, memory_demanding=False):
return afffe_baseline(annotation_type, fixed_positions) 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 ...@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet100
from bob.bio.face.utils import lookup_config_from_database 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") locals().get("database")
) )
def load(annotation_type, fixed_positions=None): def load(annotation_type, fixed_positions=None, memory_demanding=False):
return iresnet100(annotation_type, fixed_positions) 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 ...@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet34
from bob.bio.face.utils import lookup_config_from_database 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") locals().get("database")
) )
def load(annotation_type, fixed_positions=None): def load(annotation_type, fixed_positions=None, memory_demanding=False):
return iresnet34(annotation_type, fixed_positions) 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 ...@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet50
from bob.bio.face.utils import lookup_config_from_database 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") locals().get("database")
) )
def load(annotation_type, fixed_positions=None): def load(annotation_type, fixed_positions=None, memory_demanding=False):
return iresnet50(annotation_type, fixed_positions) 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