├── .gitignore ├── README.md ├── architecture.png ├── env_configs └── env_config.json ├── requirements.txt ├── runs └── default │ └── default.json ├── sample_scripts ├── dae_json.py ├── file_stats.py ├── simple_inference.py └── simple_inference_nli_init.py ├── src ├── __init__.py ├── datasets │ ├── __init__.py │ ├── data.py │ └── preprocess.py ├── models │ ├── __init__.py │ ├── attention.py │ ├── autoencode.py │ ├── decoders.py │ ├── encoders.py │ ├── generators.py │ ├── length_control.py │ ├── nli.py │ └── top.py ├── runners │ ├── __init__.py │ ├── dae_train.py │ └── inference.py ├── tests │ ├── __init__.py │ ├── test_conf.py │ └── test_data.py └── utils │ ├── __init__.py │ ├── conf.py │ ├── devices.py │ ├── io.py │ ├── logs.py │ ├── misc.py │ └── operations.py └── test_data ├── test_vecs.txt ├── test_wmt16_en.txt └── test_wmt16_en_vocabulary.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/README.md -------------------------------------------------------------------------------- /architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/architecture.png -------------------------------------------------------------------------------- /env_configs/env_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/env_configs/env_config.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch==0.3.1 2 | numpy 3 | pandas 4 | attrs 5 | -------------------------------------------------------------------------------- /runs/default/default.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/runs/default/default.json -------------------------------------------------------------------------------- /sample_scripts/dae_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/sample_scripts/dae_json.py -------------------------------------------------------------------------------- /sample_scripts/file_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/sample_scripts/file_stats.py -------------------------------------------------------------------------------- /sample_scripts/simple_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/sample_scripts/simple_inference.py -------------------------------------------------------------------------------- /sample_scripts/simple_inference_nli_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/sample_scripts/simple_inference_nli_init.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datasets/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/datasets/data.py -------------------------------------------------------------------------------- /src/datasets/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/datasets/preprocess.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/attention.py -------------------------------------------------------------------------------- /src/models/autoencode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/autoencode.py -------------------------------------------------------------------------------- /src/models/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/decoders.py -------------------------------------------------------------------------------- /src/models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/encoders.py -------------------------------------------------------------------------------- /src/models/generators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/generators.py -------------------------------------------------------------------------------- /src/models/length_control.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/length_control.py -------------------------------------------------------------------------------- /src/models/nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/nli.py -------------------------------------------------------------------------------- /src/models/top.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/models/top.py -------------------------------------------------------------------------------- /src/runners/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/runners/dae_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/runners/dae_train.py -------------------------------------------------------------------------------- /src/runners/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/runners/inference.py -------------------------------------------------------------------------------- /src/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/tests/test_conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/tests/test_conf.py -------------------------------------------------------------------------------- /src/tests/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/tests/test_data.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/utils/conf.py -------------------------------------------------------------------------------- /src/utils/devices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/utils/devices.py -------------------------------------------------------------------------------- /src/utils/io.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/utils/logs.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/utils/misc.py -------------------------------------------------------------------------------- /src/utils/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/src/utils/operations.py -------------------------------------------------------------------------------- /test_data/test_vecs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/test_data/test_vecs.txt -------------------------------------------------------------------------------- /test_data/test_wmt16_en.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/test_data/test_wmt16_en.txt -------------------------------------------------------------------------------- /test_data/test_wmt16_en_vocabulary.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zphang/usc_dae/HEAD/test_data/test_wmt16_en_vocabulary.txt --------------------------------------------------------------------------------