From 4e2e9722f91793f7d1d25c862a46c45e9448cce9 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 15 Jun 2021 18:59:27 +0200
Subject: [PATCH] Add memory demanding to pytorch models
---
bob/bio/face/config/baseline/afffe.py | 8 ++++----
bob/bio/face/config/baseline/iresnet100.py | 8 ++++----
bob/bio/face/config/baseline/iresnet34.py | 8 ++++----
bob/bio/face/config/baseline/iresnet50.py | 8 ++++----
4 files changed, 16 insertions(+), 16 deletions(-)
diff --git a/bob/bio/face/config/baseline/afffe.py b/bob/bio/face/config/baseline/afffe.py
index 840ae58e..d46eb007 100644
--- a/bob/bio/face/config/baseline/afffe.py
+++ b/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)
diff --git a/bob/bio/face/config/baseline/iresnet100.py b/bob/bio/face/config/baseline/iresnet100.py
index 6148d162..666e6f34 100644
--- a/bob/bio/face/config/baseline/iresnet100.py
+++ b/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)
diff --git a/bob/bio/face/config/baseline/iresnet34.py b/bob/bio/face/config/baseline/iresnet34.py
index f66b2b88..d8cdf0ba 100644
--- a/bob/bio/face/config/baseline/iresnet34.py
+++ b/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)
diff --git a/bob/bio/face/config/baseline/iresnet50.py b/bob/bio/face/config/baseline/iresnet50.py
index ae256242..5a33a734 100644
--- a/bob/bio/face/config/baseline/iresnet50.py
+++ b/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)
--
GitLab