diff options
author | TharinduDR <rhtdranasinghe@gmail.com> | 2021-05-24 19:30:20 +0300 |
---|---|---|
committer | TharinduDR <rhtdranasinghe@gmail.com> | 2021-05-24 19:30:20 +0300 |
commit | 61eceb744ee9508af091bb4cc5afcc98be44d7cf (patch) | |
tree | 4cb3180dc778760216d1f3c91abcb2f7bf51b16f | |
parent | 4fbb3aea38c20246c2bb7f33160940aad7aae782 (diff) |
057: Code Refactoring - Siamese Architectures
-rw-r--r-- | docs/architectures/sentence_level_architectures.md | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/docs/architectures/sentence_level_architectures.md b/docs/architectures/sentence_level_architectures.md index 77bd1dd..f753aac 100644 --- a/docs/architectures/sentence_level_architectures.md +++ b/docs/architectures/sentence_level_architectures.md @@ -62,7 +62,7 @@ Initiate and train the model like in the following code. train_df and eval_df ar from transquest.algo.sentence_level.siamesetransquest.run_model import SiameseTransQuestModel -model = SiameseTransQuestModel(MODEL_NAME, args=siamesetransquest_config) +model = SiameseTransQuestModel("xlm-roberta-large", args=siamesetransquest_config) model.train_model(train_df, eval_df) ``` |