Skip to content

Commit

Permalink
typo: torctext -> torchtext
Browse files Browse the repository at this point in the history
  • Loading branch information
raj-magesh authored Apr 19, 2023
1 parent 167dc69 commit 47afd59
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchtext/models/roberta/bundler.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ def get_model(
freeze_encoder: bool = False,
dl_kwargs: Dict[str, Any] = None,
) -> RobertaModel:
r"""get_model(head: Optional[torch.nn.Module] = None, load_weights: bool = True, freeze_encoder: bool = False, *, dl_kwargs=None) -> torctext.models.RobertaModel
r"""get_model(head: Optional[torch.nn.Module] = None, load_weights: bool = True, freeze_encoder: bool = False, *, dl_kwargs=None) -> torchtext.models.RobertaModel
Args:
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``)
Expand Down

0 comments on commit 47afd59

Please sign in to comment.