Skip to content

Commit

Permalink
Add expected installation and run times
Browse files Browse the repository at this point in the history
  • Loading branch information
samgelman committed Apr 3, 2024
1 parent f13a3f7 commit c128131
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@ conda env create -f environment.yml
conda activate metl
```

Installation typically takes approximately 5 minutes.

For GPU support, make sure you have the appropriate CUDA version installed.
Add `cudatoolkit` to the `environment.yml` file before creating the conda environment.

Expand Down Expand Up @@ -53,6 +55,8 @@ python code/train_source_model.py @args/pretrain_avgfp_local.txt --max_epochs 5
The test metrics are expected to show poor performance after such a short training run.
For instance, `pearson_total_score` may be around 0.24.

Running the limited pretraining demo takes approximately 5 minutes on CPU.

See the help message for an explanation of all the arguments
```bash
python code/train_source_model.py --help
Expand All @@ -77,6 +81,8 @@ python code/train_target_model.py @args/finetune_avgfp_local.txt --enable_progre

Following the short demonstration pretraining and finetuning process is expected to give test set Spearman correlation around 0.6.

Running the finetuning demo takes approximately 7 minutes on CPU.

See the help message for an explanation of all the arguments
```bash
python code/code/train_target_model.py --help
Expand Down

0 comments on commit c128131

Please sign in to comment.