Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit a60eaee

Browse files
committed
fixing minor docstring
1 parent 8b4048a commit a60eaee

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchtext/models/roberta/bundler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,8 @@ def get_model(self,
8484
8585
Args:
8686
head (nn.Module): A module to be attached to the encoder to perform specific task. If provided, it will replace the default member head (Default: ``None``)
87-
freeze_encoder (bool): Indicates whether to freeze the encoder weights. (Default: ``False``)
87+
load_weights (bool): Indicates whether or not to load weights if available. (Default: ``True``)
88+
freeze_encoder (bool): Indicates whether or not to freeze the encoder weights. (Default: ``False``)
8889
dl_kwargs (dictionary of keyword arguments): Passed to :func:`torch.hub.load_state_dict_from_url`. (Default: ``None``)
8990
"""
9091

0 commit comments

Comments
 (0)