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

Initial update of the doc-string of MultiNetPatchExtractor

parent c2290e77
Branches master
No related tags found
1 merge request!5Remove the need for config in extractor and algo
Pipeline #26886 passed
......@@ -37,31 +37,17 @@ class MultiNetPatchExtractor(Extractor, object):
Attributes
-----------
config_file: str
Relative name of the config file.
The path should be relative to ``config_group``,
for example: "test_data/multi_net_patch_extractor_test_config.py".
This file **must** contain at least the following definitions:
transform : object
Function namely ``transform``, which is a Compose transformation of
torchvision package, to be applied to the input samples.
A ``Network`` class, defining your network architecture. Note, if your
class is named differently, import it as ``Network``, for example:
``from bob.learn.pytorch.architectures import MyNetwork as Network``
Optional: ``network_kwargs`` to be used for ``Network`` initialization.
For example, if you want to use the latent embeddings of the autoencoder
class, set the kwargs accodingly. Note: in current extractor the
``forward()`` method of the ``Network`` is used for feature extraction.
config_group: str
Group/package name containing the configuration file. Usually all
configs should be stored in this folder/place.
For example: "bob.ip.pytorch_extractor".
Both ``config_file`` and ``config_group`` are used to access the
configuration module.
network : object
An instance of the Network to be used for feature extraction.
Note: in current extractor the ``forward()`` method of the Network
is used for feature extraction. For example, if you want to use the
latent embeddings of the autoencoder class, initialize the network
accordingly.
model_file : [str]
A list of paths to the model files to be used for network initialization.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment