You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Sep 10, 2025. It is now read-only.
Copy file name to clipboardExpand all lines: torchtext/models/roberta/bundler.py
+2-1Lines changed: 2 additions & 1 deletion
Original file line number
Diff line number
Diff line change
@@ -84,7 +84,8 @@ def get_model(self,
84
84
85
85
Args:
86
86
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``)
88
89
dl_kwargs (dictionary of keyword arguments): Passed to :func:`torch.hub.load_state_dict_from_url`. (Default: ``None``)
0 commit comments