diff --git a/docs/finetune/regression.md b/docs/finetune/regression.md index 249d0ed2..ead070b8 100644 --- a/docs/finetune/regression.md +++ b/docs/finetune/regression.md @@ -189,4 +189,4 @@ python finetune/regression/preprocess_data.py \ ### Train the model ```bash python finetune/regression/regression.py fit --config configs/regression_biomasters.yaml - ``` \ No newline at end of file + ``` diff --git a/docs/finetune/segment.md b/docs/finetune/segment.md index b7a1a043..65d73202 100644 --- a/docs/finetune/segment.md +++ b/docs/finetune/segment.md @@ -1,7 +1,7 @@ # Segmentation head fine tuning We have built an example for training a segmentation head on top of -feature maps extracted from the frozen Clay encoder. +feature maps extracted from the frozen Clay encoder. All the code for this example can be found in the [segment finetuning folder](https://github.com/Clay-foundation/model/blob/main/finetune/segment)