Skip to content

Commit

Permalink
Update multi-task-training-pt.md
Browse files Browse the repository at this point in the history
  • Loading branch information
iProzd committed Mar 2, 2024
1 parent aa5a35a commit 4716032
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion doc/train/multi-task-training-pt.md
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,11 @@ which are the same as the definition in single-task training {ref}`training_data
You can specify any positive real number weight for each task. The higher the weight, the higher the probability of being sampled in each training.
This setting is optional, and if not set, tasks will be sampled with equal weights.

An example input for multi-task training two models in water system can be found here: [multi-task input on water](../../examples/water_multi_task/pytorch_example/input_torch.json).
An example input for multi-task training two models in water system is shown as following:
```{literalinclude} ../../examples/water_multi_task/pytorch_example/input_torch.json
:language: json
:linenos:
```

## Finetune from the pretrained multi-task model

Expand Down

0 comments on commit 4716032

Please sign in to comment.