Use Weights & Biases for logging of training experiments. This project template uses the IMDB Movie Review Dataset and includes two workflows: log for training a simple text classification model and logging the results to Weights & Biases (works out-of-the-box and only requires the [training.logger] to be set in the config) and parameter-search for running a hyperparameter search using Weights & Biases Sweeps, running the experiments and logging the results.
The project.yml defines the data assets required by the
project, as well as the available commands and workflows. For details, see the
Weasel documentation.
The following commands are defined by the project. They
can be executed using weasel run [name].
Commands are only re-run if their inputs have changed.
| Command | Description |
|---|---|
login |
Log in to Weights & Biases |
data |
Extract the gold-standard annotations |
train |
Train a model using the default config |
train-search |
Run customized training runs for hyperparameter search using Weights & Biases Sweeps |
clean |
Remove intermediate files. |
The following workflows are defined by the project. They
can be executed using weasel run [name]
and will run the specified commands in order. Commands are only re-run if their
inputs have changed.
| Workflow | Steps |
|---|---|
log |
data → train |
parameter-search |
data → train-search |
The following assets are defined by the project. They can
be fetched by running weasel assets
in the project directory.
| File | Source | Description |
|---|---|---|
assets/aclImdb_v1.tar.gz |
URL | Movie Review Dataset for sentiment analysis by Maas et al., ACL 2011. |