├── .gitignore ├── README.md ├── data ├── test.ape.json ├── train.ape.json └── valid.ape.json ├── images ├── 15759776402677.jpg └── 15759784507903.jpg └── onmt ├── __init__.py ├── decoders ├── __init__.py ├── cnn_decoder.py ├── decoder.py ├── ensemble.py └── transformer.py ├── encoders ├── __init__.py ├── audio_encoder.py ├── cnn_encoder.py ├── encoder.py ├── image_encoder.py ├── mean_encoder.py ├── rnn_encoder.py └── transformer.py ├── inputters ├── __init__.py ├── audio_dataset.py ├── datareader_base.py ├── dataset_base.py ├── image_dataset.py ├── inputter.py ├── text_dataset.py └── vec_dataset.py ├── model_builder.py ├── models ├── __init__.py ├── model.py ├── model_saver.py ├── sru.py └── stacked_rnn.py ├── modules ├── __init__.py ├── average_attn.py ├── conv_multi_step_attention.py ├── copy_generator.py ├── embeddings.py ├── gate.py ├── global_attention.py ├── multi_headed_attn.py ├── position_ffn.py ├── sparse_activations.py ├── sparse_losses.py ├── structured_attention.py ├── util_class.py └── weight_norm.py ├── opts.py ├── tests ├── __init__.py ├── output_hyp.txt ├── pull_request_chk.sh ├── rebuild_test_models.sh ├── sample_glove.txt ├── test_attention.py ├── test_audio_dataset.py ├── test_beam.py ├── test_beam_search.py ├── test_copy_generator.py ├── test_embeddings.py ├── test_image_dataset.py ├── test_model2.pt ├── test_models.py ├── test_models.sh ├── test_preprocess.py ├── test_random_sampling.py ├── test_simple.py ├── test_structured_attention.py ├── test_text_dataset.py ├── test_translation_server.py └── utils_for_tests.py ├── train_single.py ├── trainer.py ├── translate ├── __init__.py ├── beam.py ├── beam_search.py ├── decode_strategy.py ├── penalties.py ├── random_sampling.py ├── translation.py ├── translation_server.py └── translator.py └── utils ├── __init__.py ├── cnn_factory.py ├── distributed.py ├── earlystopping.py ├── logging.py ├── loss.py ├── misc.py ├── optimizers.py ├── parse.py ├── report_manager.py ├── rnn_factory.py └── statistics.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | *.org 3 | 4 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/README.md -------------------------------------------------------------------------------- /data/test.ape.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/data/test.ape.json -------------------------------------------------------------------------------- /data/train.ape.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/data/train.ape.json -------------------------------------------------------------------------------- /data/valid.ape.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/data/valid.ape.json -------------------------------------------------------------------------------- /images/15759776402677.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/images/15759776402677.jpg -------------------------------------------------------------------------------- /images/15759784507903.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/images/15759784507903.jpg -------------------------------------------------------------------------------- /onmt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/__init__.py -------------------------------------------------------------------------------- /onmt/decoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/decoders/__init__.py -------------------------------------------------------------------------------- /onmt/decoders/cnn_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/decoders/cnn_decoder.py -------------------------------------------------------------------------------- /onmt/decoders/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/decoders/decoder.py -------------------------------------------------------------------------------- /onmt/decoders/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/decoders/ensemble.py -------------------------------------------------------------------------------- /onmt/decoders/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/decoders/transformer.py -------------------------------------------------------------------------------- /onmt/encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/__init__.py -------------------------------------------------------------------------------- /onmt/encoders/audio_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/audio_encoder.py -------------------------------------------------------------------------------- /onmt/encoders/cnn_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/cnn_encoder.py -------------------------------------------------------------------------------- /onmt/encoders/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/encoder.py -------------------------------------------------------------------------------- /onmt/encoders/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/image_encoder.py -------------------------------------------------------------------------------- /onmt/encoders/mean_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/mean_encoder.py -------------------------------------------------------------------------------- /onmt/encoders/rnn_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/rnn_encoder.py -------------------------------------------------------------------------------- /onmt/encoders/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/encoders/transformer.py -------------------------------------------------------------------------------- /onmt/inputters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/__init__.py -------------------------------------------------------------------------------- /onmt/inputters/audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/audio_dataset.py -------------------------------------------------------------------------------- /onmt/inputters/datareader_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/datareader_base.py -------------------------------------------------------------------------------- /onmt/inputters/dataset_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/dataset_base.py -------------------------------------------------------------------------------- /onmt/inputters/image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/image_dataset.py -------------------------------------------------------------------------------- /onmt/inputters/inputter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/inputter.py -------------------------------------------------------------------------------- /onmt/inputters/text_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/text_dataset.py -------------------------------------------------------------------------------- /onmt/inputters/vec_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/inputters/vec_dataset.py -------------------------------------------------------------------------------- /onmt/model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/model_builder.py -------------------------------------------------------------------------------- /onmt/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/models/__init__.py -------------------------------------------------------------------------------- /onmt/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/models/model.py -------------------------------------------------------------------------------- /onmt/models/model_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/models/model_saver.py -------------------------------------------------------------------------------- /onmt/models/sru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/models/sru.py -------------------------------------------------------------------------------- /onmt/models/stacked_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/models/stacked_rnn.py -------------------------------------------------------------------------------- /onmt/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/__init__.py -------------------------------------------------------------------------------- /onmt/modules/average_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/average_attn.py -------------------------------------------------------------------------------- /onmt/modules/conv_multi_step_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/conv_multi_step_attention.py -------------------------------------------------------------------------------- /onmt/modules/copy_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/copy_generator.py -------------------------------------------------------------------------------- /onmt/modules/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/embeddings.py -------------------------------------------------------------------------------- /onmt/modules/gate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/gate.py -------------------------------------------------------------------------------- /onmt/modules/global_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/global_attention.py -------------------------------------------------------------------------------- /onmt/modules/multi_headed_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/multi_headed_attn.py -------------------------------------------------------------------------------- /onmt/modules/position_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/position_ffn.py -------------------------------------------------------------------------------- /onmt/modules/sparse_activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/sparse_activations.py -------------------------------------------------------------------------------- /onmt/modules/sparse_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/sparse_losses.py -------------------------------------------------------------------------------- /onmt/modules/structured_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/structured_attention.py -------------------------------------------------------------------------------- /onmt/modules/util_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/util_class.py -------------------------------------------------------------------------------- /onmt/modules/weight_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/modules/weight_norm.py -------------------------------------------------------------------------------- /onmt/opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/opts.py -------------------------------------------------------------------------------- /onmt/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /onmt/tests/output_hyp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/output_hyp.txt -------------------------------------------------------------------------------- /onmt/tests/pull_request_chk.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/pull_request_chk.sh -------------------------------------------------------------------------------- /onmt/tests/rebuild_test_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/rebuild_test_models.sh -------------------------------------------------------------------------------- /onmt/tests/sample_glove.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/sample_glove.txt -------------------------------------------------------------------------------- /onmt/tests/test_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_attention.py -------------------------------------------------------------------------------- /onmt/tests/test_audio_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_audio_dataset.py -------------------------------------------------------------------------------- /onmt/tests/test_beam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_beam.py -------------------------------------------------------------------------------- /onmt/tests/test_beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_beam_search.py -------------------------------------------------------------------------------- /onmt/tests/test_copy_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_copy_generator.py -------------------------------------------------------------------------------- /onmt/tests/test_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_embeddings.py -------------------------------------------------------------------------------- /onmt/tests/test_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_image_dataset.py -------------------------------------------------------------------------------- /onmt/tests/test_model2.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_model2.pt -------------------------------------------------------------------------------- /onmt/tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_models.py -------------------------------------------------------------------------------- /onmt/tests/test_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_models.sh -------------------------------------------------------------------------------- /onmt/tests/test_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_preprocess.py -------------------------------------------------------------------------------- /onmt/tests/test_random_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_random_sampling.py -------------------------------------------------------------------------------- /onmt/tests/test_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_simple.py -------------------------------------------------------------------------------- /onmt/tests/test_structured_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_structured_attention.py -------------------------------------------------------------------------------- /onmt/tests/test_text_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_text_dataset.py -------------------------------------------------------------------------------- /onmt/tests/test_translation_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/test_translation_server.py -------------------------------------------------------------------------------- /onmt/tests/utils_for_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/tests/utils_for_tests.py -------------------------------------------------------------------------------- /onmt/train_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/train_single.py -------------------------------------------------------------------------------- /onmt/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/trainer.py -------------------------------------------------------------------------------- /onmt/translate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/__init__.py -------------------------------------------------------------------------------- /onmt/translate/beam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/beam.py -------------------------------------------------------------------------------- /onmt/translate/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/beam_search.py -------------------------------------------------------------------------------- /onmt/translate/decode_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/decode_strategy.py -------------------------------------------------------------------------------- /onmt/translate/penalties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/penalties.py -------------------------------------------------------------------------------- /onmt/translate/random_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/random_sampling.py -------------------------------------------------------------------------------- /onmt/translate/translation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/translation.py -------------------------------------------------------------------------------- /onmt/translate/translation_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/translation_server.py -------------------------------------------------------------------------------- /onmt/translate/translator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/translate/translator.py -------------------------------------------------------------------------------- /onmt/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/__init__.py -------------------------------------------------------------------------------- /onmt/utils/cnn_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/cnn_factory.py -------------------------------------------------------------------------------- /onmt/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/distributed.py -------------------------------------------------------------------------------- /onmt/utils/earlystopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/earlystopping.py -------------------------------------------------------------------------------- /onmt/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/logging.py -------------------------------------------------------------------------------- /onmt/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/loss.py -------------------------------------------------------------------------------- /onmt/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/misc.py -------------------------------------------------------------------------------- /onmt/utils/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/optimizers.py -------------------------------------------------------------------------------- /onmt/utils/parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/parse.py -------------------------------------------------------------------------------- /onmt/utils/report_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/report_manager.py -------------------------------------------------------------------------------- /onmt/utils/rnn_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/rnn_factory.py -------------------------------------------------------------------------------- /onmt/utils/statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Chenny0808/ape210k/HEAD/onmt/utils/statistics.py --------------------------------------------------------------------------------