Skip to content

Commit 3d764a3

Browse files
Myle Ottfacebook-github-bot
Myle Ott
authored andcommitted
Update torch.hub usage
Summary: Pull Request resolved: fairinternal/fairseq-py#770 Differential Revision: D16491911 Pulled By: myleott fbshipit-source-id: 8dd2b76f8fa24183640ae9d1129ea47ded77d43d
1 parent b49ea81 commit 3d764a3

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

examples/backtranslation/README.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -13,14 +13,14 @@ Transformer <br> ([Edunov et al., 2018](https://arxiv.org/abs/1808.09381); WMT'1
1313
Interactive generation from the full ensemble via PyTorch Hub:
1414
```
1515
>>> import torch
16+
>>> torch.hub.list('pytorch/fairseq')
17+
[..., 'transformer.wmt14.en-fr', 'transformer.wmt16.en-de', 'transformer.wmt18.en-de', ... ]
1618
>>> en2de_ensemble = torch.hub.load(
1719
... 'pytorch/fairseq',
18-
... 'transformer',
19-
... model_name_or_path='transformer.wmt18.en-de',
20+
... 'transformer.wmt18.en-de',
2021
... checkpoint_file='wmt18.model1.pt:wmt18.model2.pt:wmt18.model3.pt:wmt18.model4.pt:wmt18.model5.pt',
2122
... data_name_or_path='.',
2223
... tokenizer='moses',
23-
... aggressive_dash_splits=True,
2424
... bpe='subword_nmt',
2525
... )
2626
>>> len(en2de_ensemble.models)

examples/language_model/README.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,13 @@ Adaptive Inputs <br> ([Baevski and Auli, 2018](https://arxiv.org/abs/1809.10853)
1313
Interactive generation via PyTorch Hub:
1414
```
1515
>>> import torch
16+
>>> torch.hub.list('pytorch/fairseq')
17+
[..., 'transformer_lm.gbw.adaptive_huge', 'transformer_lm.wiki103.adaptive', ...]
1618
>>> lm = torch.hub.load(
1719
... 'pytorch/fairseq',
18-
... 'transformer_lm',
19-
... model_name_or_path='transformer_lm.wiki103.adaptive',
20+
... 'transformer_lm.wiki103.adaptive',
2021
... data_name_or_path='./data-bin',
2122
... tokenizer='moses',
22-
... aggressive_dash_splits=True,
2323
... no_escape=True,
2424
... beam=1,
2525
... sampling=True,

examples/translation/README.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,13 @@ Transformer <br> ([Edunov et al., 2018](https://arxiv.org/abs/1808.09381); WMT'1
1616
Interactive generation via PyTorch Hub:
1717
```
1818
>>> import torch
19+
>>> torch.hub.list('pytorch/fairseq')
20+
[..., 'transformer.wmt14.en-fr', 'transformer.wmt16.en-de', 'transformer.wmt18.en-de', ... ]
1921
>>> en2de = torch.hub.load(
2022
... 'pytorch/fairseq',
21-
... 'transformer',
22-
... model_name_or_path='transformer.wmt16.en-de',
23+
... 'transformer.wmt16.en-de',
2324
... data_name_or_path='.',
2425
... tokenizer='moses',
25-
... aggressive_dash_splits=True,
2626
... bpe='subword_nmt',
2727
... )
2828
>>> print(en2de.models[0].__class__)

0 commit comments

Comments
 (0)