... | ... | @@ -5,7 +5,7 @@ All the following section subtitles are relative to paths starting from this dir |
|
|
## run.py
|
|
|
The starting point of the model. It accepts these command-line arguments:
|
|
|
|
|
|
* *--config* → path to the [configuration file](https://gitlab.fbk.eu/dsip/templates/dl_setup/-/wikis/Configuration-file);
|
|
|
* *--config* → path to the [configuration file](Configuration-file);
|
|
|
* *--train* → flag to train model;
|
|
|
* *--inference* → flag to use model for inference.
|
|
|
|
... | ... | @@ -32,7 +32,7 @@ The train_model function accepts these required parameters: |
|
|
* *epochs* → number of epochs;
|
|
|
* *data_loaders* → dictionary. Keys are strings representing phases of the training process (train, validation, test). Values are DataLoader instances;
|
|
|
* *device* → device that will be used;
|
|
|
* *net_weights_dir* → path to the directory in which model weights will be saved (loaded from [configuration file](https://gitlab.fbk.eu/dsip/templates/dl_setup/-/wikis/Configuration-file) in train function inside run.py file)
|
|
|
* *net_weights_dir* → path to the directory in which model weights will be saved (loaded from [configuration file](Configuration-file) in train function inside run.py file)
|
|
|
|
|
|
There are also some optional parameters:
|
|
|
|
... | ... | @@ -59,7 +59,7 @@ To summarize: |
|
|
* it avoids overfitting saving the best validation model;
|
|
|
* it saves training time stopping the loop before the end.
|
|
|
|
|
|
Net weights will be saved inside the directory specified inside [configuration file](https://gitlab.fbk.eu/dsip/templates/dl_setup/-/wikis/Configuration-file). The dir structure will be:
|
|
|
Net weights will be saved inside the directory specified inside [configuration file](Configuration-file). The dir structure will be:
|
|
|
```
|
|
|
net_weights_dir
|
|
|
├──┐experiment_name
|
... | ... | |