From 47160327fa53078036d1d4f82c02db114ca430ca Mon Sep 17 00:00:00 2001 From: Duo <50307526+iProzd@users.noreply.github.com> Date: Sat, 2 Mar 2024 15:41:03 +0800 Subject: [PATCH] Update multi-task-training-pt.md --- doc/train/multi-task-training-pt.md | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/doc/train/multi-task-training-pt.md b/doc/train/multi-task-training-pt.md index 69d7dfc529..d1d0f9cd0d 100644 --- a/doc/train/multi-task-training-pt.md +++ b/doc/train/multi-task-training-pt.md @@ -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