Commit 2da69255 authored by Olegs NIKISINS's avatar Olegs NIKISINS

Renamed the script used for AE training, both in setup.py and docs

parent 61202ef7
Pipeline #26277 passed with stage
in 7 minutes and 15 seconds
......@@ -9,7 +9,7 @@ As an example, to train an autoencoder on facial images extracted from the Celeb
.. code-block:: sh
./bin/pytorch-train-autoencoder-pad.py \ # script used for autoencoders training, can be used for other networks as-well
./bin/train_autoencoder.py \ # script used for autoencoders training, can be used for other networks as-well
<FOLDER_CONTAINING_TRAINING_DATA> \ # substitute the path pointing to training data
<FOLDER_TO_SAVE_THE_RESULTS>/ \ # substitute the path to save the results to
-c autoencoder/net1_celeba.py \ # configuration file defining the AE, database, and training parameters
......@@ -24,7 +24,7 @@ People in Idiap can benefit from GPU cluster, running the training as follows:
--name <NAME_OF_EXPERIMENT> \ # define the name of th job (Idiap only)
--log-dir <FOLDER_TO_SAVE_THE_RESULTS>/logs/ \ # substitute the path to save the logs to (Idiap only)
--environment="PYTHONUNBUFFERED=1" -- \ #
./bin/pytorch-train-autoencoder-pad.py \ # script used for autoencoders training, cand be used for other networks as-well
./bin/train_autoencoder.py \ # script used for autoencoders training, cand be used for other networks as-well
<FOLDER_CONTAINING_TRAINING_DATA> \ # substitute the path pointing to training data
<FOLDER_TO_SAVE_THE_RESULTS>/ \ # substitute the path to save the results to
-c autoencoder/net1_celeba.py \ # configuration file defining the AE, database, and training parameters
......@@ -37,7 +37,7 @@ For a more detailed documentation of functionality available in the training scr
.. code-block:: sh
./bin/pytorch-train-autoencoder-pad.py --help # note: remove ./bin/ if buildout is not used
./bin/train_autoencoder.py --help # note: remove ./bin/ if buildout is not used
Please inspect the corresponding configuration file, ``net1_celeba.py`` for example, for more details on how to define the database, network architecture and training parameters.
......
......@@ -72,7 +72,7 @@ setup(
'train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main',
'train_dcgan.py = bob.learn.pytorch.scripts.train_dcgan:main',
'train_conditionalgan.py = bob.learn.pytorch.scripts.train_conditionalgan:main',
'pytorch-train-autoencoder-pad.py = bob.learn.pytorch.scripts.train_autoencoder:main',
'train_autoencoder.py = bob.learn.pytorch.scripts.train_autoencoder:main',
],
},
......
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