fairseq/tests
Liezl Puzon f766c9a0d5 Pass in kwargs and SequenceGenerator class to init BacktranslationDataset
Summary: This generalizes BacktranslationDataset to allow us to use any SequenceGenerator class. For example, if we want to use this model in PyTorch Translate, we can pass the following to BacktraanslationDataset init: (1) a PyTorch Translate SequenceGenerator class as generator_class and (2) the appropriate args for initializing that class as kwargs.

Reviewed By: xianxl

Differential Revision: D10156552

fbshipit-source-id: 0495d825bf4727da96d0d9a40dc434135ff3486c
2018-10-02 18:22:27 -07:00
..
__init__.py fairseq-py goes distributed (#106) 2018-02-27 17:09:42 -05:00
test_average_checkpoints.py Merge internal changes (#136) 2018-04-02 10:13:07 -04:00
test_backtranslation_dataset.py Pass in kwargs and SequenceGenerator class to init BacktranslationDataset 2018-10-02 18:22:27 -07:00
test_binaries.py Better support for various c10d API changes 2018-09-25 17:36:43 -04:00
test_character_token_embedder.py fix tests 2018-09-03 19:15:23 -04:00
test_convtbc.py Remove more Variable() calls (#198) 2018-06-25 12:23:04 -04:00
test_dictionary.py Remove more Variable() calls (#198) 2018-06-25 12:23:04 -04:00
test_iterators.py Further generalize EpochBatchIterator and move iterators into new file 2018-09-03 19:15:23 -04:00
test_label_smoothing.py Add FairseqTask 2018-06-15 13:05:22 -06:00
test_noising.py fbshipit-source-id: 6a835d32f9dc5e0de118f1b46d365d0e0cc85e11 2018-09-30 12:28:20 -07:00
test_reproducibility.py Add unit test to verify reproducibility after reloading checkpoints 2018-09-25 17:36:43 -04:00
test_sequence_generator.py Online backtranslation module 2018-09-25 17:36:43 -04:00
test_sequence_scorer.py Add FairseqTask 2018-06-15 13:05:22 -06:00
test_train.py core changes to support latte collab 2018-09-25 17:36:43 -04:00
test_utils.py Remove more Variable() calls (#198) 2018-06-25 12:23:04 -04:00
utils.py Online backtranslation module 2018-09-25 17:36:43 -04:00