diff --git a/docs/transfer/transfer.md b/docs/transfer/transfer.md index be068e8..32ff94b 100644 --- a/docs/transfer/transfer.md +++ b/docs/transfer/transfer.md @@ -137,7 +137,6 @@ Similar to Borzoi training, arguments for training learning is also indicated in "transfer": { "mode": "adapter", "adapter": "locon", - "adapter_latent": 8, "conv_select": 4 }, ``` diff --git a/src/baskerville/scripts/hound_transfer.py b/src/baskerville/scripts/hound_transfer.py index 88ca008..886bb02 100755 --- a/src/baskerville/scripts/hound_transfer.py +++ b/src/baskerville/scripts/hound_transfer.py @@ -116,7 +116,7 @@ def main(): params_transfer = params["transfer"] transfer_mode = params_transfer.get("mode", "full") transfer_adapter = params_transfer.get("adapter", None) - transfer_latent = params_transfer.get("latent", 8) + transfer_latent = params_transfer.get("adapter_latent", 8) transfer_conv_select = params_transfer.get("conv_select", 4) transfer_conv_rank = params_transfer.get("conv_latent", 4) transfer_lora_alpha = params_transfer.get("lora_alpha", 16)