Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
bob
bob.bio.face
Commits
4e2e9722
Commit
4e2e9722
authored
Jun 15, 2021
by
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
Changes
4
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
bob/bio/face/config/baseline/afffe.py
View file @
4e2e9722
...
...
@@ -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
View file @
4e2e9722
...
...
@@ -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
View file @
4e2e9722
...
...
@@ -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
View file @
4e2e9722
...
...
@@ -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
)
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment