Skip to content

GitLab

Commit 097fa65b authored by Mostofa Patwary's avatar Mostofa Patwary
Browse files

fixed alignment

parent a96ad63f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
pretrain_ict.py
View file @ 097fa65b
...@@ -37,8 +37,7 @@ def pretrain_ict_model_provider(): ...@@ -37,8 +37,7 @@ def pretrain_ict_model_provider():
model = biencoder_model_provider( model = biencoder_model_provider(
only_context_model=False, only_context_model=False,
only_query_model=False, only_query_model=False,
shared_query_context_model=args.shared_query_context_model shared_query_context_model=args.shared_query_context_model)
)
return model return model
def get_group_world_size_rank(): def get_group_world_size_rank():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment