├── README.md ├── data ├── BLEU_REF.txt ├── dev.txt └── train.txt ├── requirements.txt ├── resource └── vocab.txt ├── scripts ├── evaluate.py ├── train.sh └── translate.sh ├── transformer ├── __init__.py ├── model │ ├── __init__.py │ ├── attention_layer.py │ ├── beam_search.py │ ├── beam_search_test.py │ ├── embedding_layer.py │ ├── ffn_layer.py │ ├── model_params.py │ ├── model_utils.py │ ├── model_utils_test.py │ └── transformer.py └── utils │ ├── __init__.py │ ├── compute_bleu.py │ ├── dataset.py │ ├── flags │ ├── README.md │ ├── __init__.py │ ├── _base.py │ ├── _benchmark.py │ ├── _conventions.py │ ├── _device.py │ ├── _distribution.py │ ├── _misc.py │ ├── _performance.py │ ├── core.py │ ├── flags_test.py │ └── guidelines.md │ ├── logs │ ├── __init__.py │ ├── cloud_lib.py │ ├── cloud_lib_test.py │ ├── guidelines.md │ ├── hooks.py │ ├── hooks_helper.py │ ├── hooks_helper_test.py │ ├── hooks_test.py │ ├── logger.py │ ├── logger_test.py │ ├── metric_hook.py │ ├── metric_hook_test.py │ └── mlperf_helper.py │ ├── metrics.py │ ├── misc │ ├── __init__.py │ ├── callstack_sampler.py │ ├── distribution_utils.py │ ├── distribution_utils_test.py │ ├── keras_utils.py │ ├── model_helpers.py │ ├── model_helpers_test.py │ └── tpu_lib.py │ ├── params_mapping.py │ ├── schedule.py │ ├── schedule_test.py │ ├── testing │ ├── __init__.py │ ├── integration.py │ ├── mock_lib.py │ ├── perfzero_benchmark.py │ ├── pylint.rcfile │ ├── reference_data.py │ ├── reference_data │ │ ├── reference_data_test │ │ │ ├── dense │ │ │ │ ├── expected_graph │ │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ │ ├── model.ckpt.index │ │ │ │ ├── results.json │ │ │ │ └── tf_version.json │ │ │ └── uniform_random │ │ │ │ ├── expected_graph │ │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ │ ├── model.ckpt.index │ │ │ │ ├── results.json │ │ │ │ └── tf_version.json │ │ └── resnet │ │ │ ├── batch-size-32_bottleneck_projection_version-1_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ ├── batch-size-32_bottleneck_projection_version-2_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ ├── batch-size-32_bottleneck_version-1_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ ├── batch-size-32_bottleneck_version-2_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ ├── batch-size-32_building_projection_version-1_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ ├── batch-size-32_building_projection_version-2_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ ├── batch-size-32_building_version-1_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ ├── batch-size-32_building_version-2_width-8_channels-4 │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ │ │ └── batch_norm │ │ │ ├── expected_graph │ │ │ ├── model.ckpt.data-00000-of-00001 │ │ │ ├── model.ckpt.index │ │ │ ├── results.json │ │ │ └── tf_version.json │ ├── reference_data_test.py │ └── scripts │ │ ├── builds_common.sh │ │ ├── ci_sanity.sh │ │ └── presubmit.sh │ ├── tokenizer.py │ ├── tokenizer_test.py │ └── tpu.py ├── transformer_main.py └── translate.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/README.md -------------------------------------------------------------------------------- /data/BLEU_REF.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/data/BLEU_REF.txt -------------------------------------------------------------------------------- /data/dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/data/dev.txt -------------------------------------------------------------------------------- /data/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/data/train.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tensorflow-gpu >= 1.13.0 -------------------------------------------------------------------------------- /resource/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/resource/vocab.txt -------------------------------------------------------------------------------- /scripts/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/scripts/evaluate.py -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /scripts/translate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/scripts/translate.sh -------------------------------------------------------------------------------- /transformer/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding:utf-8 -*- 2 | 3 | 4 | if __name__ == '__main__': 5 | pass -------------------------------------------------------------------------------- /transformer/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transformer/model/attention_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/attention_layer.py -------------------------------------------------------------------------------- /transformer/model/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/beam_search.py -------------------------------------------------------------------------------- /transformer/model/beam_search_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/beam_search_test.py -------------------------------------------------------------------------------- /transformer/model/embedding_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/embedding_layer.py -------------------------------------------------------------------------------- /transformer/model/ffn_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/ffn_layer.py -------------------------------------------------------------------------------- /transformer/model/model_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/model_params.py -------------------------------------------------------------------------------- /transformer/model/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/model_utils.py -------------------------------------------------------------------------------- /transformer/model/model_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/model_utils_test.py -------------------------------------------------------------------------------- /transformer/model/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/model/transformer.py -------------------------------------------------------------------------------- /transformer/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transformer/utils/compute_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/compute_bleu.py -------------------------------------------------------------------------------- /transformer/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/dataset.py -------------------------------------------------------------------------------- /transformer/utils/flags/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/README.md -------------------------------------------------------------------------------- /transformer/utils/flags/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transformer/utils/flags/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/_base.py -------------------------------------------------------------------------------- /transformer/utils/flags/_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/_benchmark.py -------------------------------------------------------------------------------- /transformer/utils/flags/_conventions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/_conventions.py -------------------------------------------------------------------------------- /transformer/utils/flags/_device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/_device.py -------------------------------------------------------------------------------- /transformer/utils/flags/_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/_distribution.py -------------------------------------------------------------------------------- /transformer/utils/flags/_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/_misc.py -------------------------------------------------------------------------------- /transformer/utils/flags/_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/_performance.py -------------------------------------------------------------------------------- /transformer/utils/flags/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/core.py -------------------------------------------------------------------------------- /transformer/utils/flags/flags_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/flags_test.py -------------------------------------------------------------------------------- /transformer/utils/flags/guidelines.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/flags/guidelines.md -------------------------------------------------------------------------------- /transformer/utils/logs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transformer/utils/logs/cloud_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/cloud_lib.py -------------------------------------------------------------------------------- /transformer/utils/logs/cloud_lib_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/cloud_lib_test.py -------------------------------------------------------------------------------- /transformer/utils/logs/guidelines.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/guidelines.md -------------------------------------------------------------------------------- /transformer/utils/logs/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/hooks.py -------------------------------------------------------------------------------- /transformer/utils/logs/hooks_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/hooks_helper.py -------------------------------------------------------------------------------- /transformer/utils/logs/hooks_helper_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/hooks_helper_test.py -------------------------------------------------------------------------------- /transformer/utils/logs/hooks_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/hooks_test.py -------------------------------------------------------------------------------- /transformer/utils/logs/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/logger.py -------------------------------------------------------------------------------- /transformer/utils/logs/logger_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/logger_test.py -------------------------------------------------------------------------------- /transformer/utils/logs/metric_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/metric_hook.py -------------------------------------------------------------------------------- /transformer/utils/logs/metric_hook_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/metric_hook_test.py -------------------------------------------------------------------------------- /transformer/utils/logs/mlperf_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/logs/mlperf_helper.py -------------------------------------------------------------------------------- /transformer/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/metrics.py -------------------------------------------------------------------------------- /transformer/utils/misc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transformer/utils/misc/callstack_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/misc/callstack_sampler.py -------------------------------------------------------------------------------- /transformer/utils/misc/distribution_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/misc/distribution_utils.py -------------------------------------------------------------------------------- /transformer/utils/misc/distribution_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/misc/distribution_utils_test.py -------------------------------------------------------------------------------- /transformer/utils/misc/keras_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/misc/keras_utils.py -------------------------------------------------------------------------------- /transformer/utils/misc/model_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/misc/model_helpers.py -------------------------------------------------------------------------------- /transformer/utils/misc/model_helpers_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/misc/model_helpers_test.py -------------------------------------------------------------------------------- /transformer/utils/misc/tpu_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/misc/tpu_lib.py -------------------------------------------------------------------------------- /transformer/utils/params_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/params_mapping.py -------------------------------------------------------------------------------- /transformer/utils/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/schedule.py -------------------------------------------------------------------------------- /transformer/utils/schedule_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/schedule_test.py -------------------------------------------------------------------------------- /transformer/utils/testing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transformer/utils/testing/integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/integration.py -------------------------------------------------------------------------------- /transformer/utils/testing/mock_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/mock_lib.py -------------------------------------------------------------------------------- /transformer/utils/testing/perfzero_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/perfzero_benchmark.py -------------------------------------------------------------------------------- /transformer/utils/testing/pylint.rcfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/pylint.rcfile -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data.py -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/dense/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/reference_data_test/dense/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/dense/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/reference_data_test/dense/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/dense/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/reference_data_test/dense/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/dense/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/reference_data_test/dense/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/dense/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180325", "v1.7.0-rc1-750-g6c1737e6c8"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/uniform_random/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/reference_data_test/uniform_random/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/uniform_random/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- 1 | ʼ|? -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/uniform_random/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/reference_data_test/uniform_random/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/uniform_random/results.json: -------------------------------------------------------------------------------- 1 | [0.9872556924819946] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/reference_data_test/uniform_random/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180325", "v1.7.0-rc1-750-g6c1737e6c8"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-1_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_projection_version-2_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-1_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_bottleneck_version-2_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-1_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_projection_version-2_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-1_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch-size-32_building_version-2_width-8_channels-4/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch_norm/expected_graph: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch_norm/expected_graph -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch_norm/model.ckpt.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch_norm/model.ckpt.data-00000-of-00001 -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch_norm/model.ckpt.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch_norm/model.ckpt.index -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch_norm/results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data/resnet/batch_norm/results.json -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data/resnet/batch_norm/tf_version.json: -------------------------------------------------------------------------------- 1 | ["1.8.0-dev20180408", "v1.7.0-1345-gb874783ccd"] -------------------------------------------------------------------------------- /transformer/utils/testing/reference_data_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/reference_data_test.py -------------------------------------------------------------------------------- /transformer/utils/testing/scripts/builds_common.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/scripts/builds_common.sh -------------------------------------------------------------------------------- /transformer/utils/testing/scripts/ci_sanity.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/scripts/ci_sanity.sh -------------------------------------------------------------------------------- /transformer/utils/testing/scripts/presubmit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/testing/scripts/presubmit.sh -------------------------------------------------------------------------------- /transformer/utils/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/tokenizer.py -------------------------------------------------------------------------------- /transformer/utils/tokenizer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/tokenizer_test.py -------------------------------------------------------------------------------- /transformer/utils/tpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer/utils/tpu.py -------------------------------------------------------------------------------- /transformer_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/transformer_main.py -------------------------------------------------------------------------------- /translate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liu-nlper/dialogue-utterance-rewriter/HEAD/translate.py --------------------------------------------------------------------------------