Skip to content
Snippets Groups Projects
Commit 150166c7 authored by Olegs NIKISINS's avatar Olegs NIKISINS
Browse files

Renamed the test config for MultiNetPatchExtractor, updated docstrings accordingly

parent 55016a19
Branches
No related tags found
1 merge request!4MLPAlgorithm PAD algorithm V1 version
Pipeline #26675 passed
...@@ -40,7 +40,7 @@ class MultiNetPatchExtractor(Extractor, object): ...@@ -40,7 +40,7 @@ class MultiNetPatchExtractor(Extractor, object):
config_file: str config_file: str
Relative name of the config file. Relative name of the config file.
The path should be relative to ``config_group``, The path should be relative to ``config_group``,
for example: "autoencoder/net1_batl_3_layers_partial.py". for example: "test_data/multi_net_patch_extractor_test_config.py".
This file **must** contain at least the following definitions: This file **must** contain at least the following definitions:
...@@ -59,7 +59,7 @@ class MultiNetPatchExtractor(Extractor, object): ...@@ -59,7 +59,7 @@ class MultiNetPatchExtractor(Extractor, object):
config_group: str config_group: str
Group/package name containing the configuration file. Usually all Group/package name containing the configuration file. Usually all
configs should be stored in this folder/place. configs should be stored in this folder/place.
For example: "bob.learn.pytorch.config". For example: "bob.ip.pytorch_extractor".
Both ``config_file`` and ``config_group`` are used to access the Both ``config_file`` and ``config_group`` are used to access the
configuration module. configuration module.
......
...@@ -65,7 +65,7 @@ def test_multi_net_patch_extractor(): ...@@ -65,7 +65,7 @@ def test_multi_net_patch_extractor():
# ========================================================================= # =========================================================================
# test the extractor: # test the extractor:
CONFIG_FILE = "test_data/net1_test_config.py" # config containing an instance of Composed Transform and a Network class to be used in feature extractor CONFIG_FILE = "test_data/multi_net_patch_extractor_test_config.py" # config containing an instance of Composed Transform and a Network class to be used in feature extractor
CONFIG_GROUP = "bob.ip.pytorch_extractor" CONFIG_GROUP = "bob.ip.pytorch_extractor"
# use specific/unique model for each patch. Models pre-trained on CelebA and fine-tuned (3 layers) on BATL: # use specific/unique model for each patch. Models pre-trained on CelebA and fine-tuned (3 layers) on BATL:
......
...@@ -8,8 +8,6 @@ ...@@ -8,8 +8,6 @@
from torchvision import transforms from torchvision import transforms
#from bob.pad.face.database import CELEBAPadDatabase
from torch import nn from torch import nn
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment