Tacotron2+Waveglow/PyT * AMP support * Data preprocessing for Tacotron 2 training * Fixed dropouts on LSTMCells SSD/PyT * script and notebook for inference * AMP support * README update * updates to examples/* BERT/PyT * initial release GNMT/PyT * Default container updated to NGC PyTorch 19.05-py3 * Mixed precision training implemented using APEX AMP * Added inference throughput and latency results on NVIDIA Tesla V100 16G * Added option to run inference on user-provided raw input text from command line NCF/PyT * Updated performance tables. * Default container changed to PyTorch 19.06-py3. * Caching validation negatives between runs Transformer/PyT * new README * jit support added UNet Medical/TF * inference example scripts added * inference benchmark measuring latency added * TRT/TF-TRT support added * README updated GNMT/TF * Performance improvements Small updates (mostly README) for other models.
35 lines
1.1 KiB
Bash
35 lines
1.1 KiB
Bash
# Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
DATASET_DIR=/data/wmt14_en_de_joined_dict
|
|
TEXT=examples/translation/wmt14_en_de
|
|
|
|
(
|
|
cd examples/translation
|
|
bash prepare-wmt14en2de.sh --scaling18
|
|
)
|
|
|
|
python preprocess.py \
|
|
--source-lang en \
|
|
--target-lang de \
|
|
--trainpref $TEXT/train \
|
|
--validpref $TEXT/valid \
|
|
--testpref $TEXT/test \
|
|
--destdir ${DATASET_DIR} \
|
|
--nwordssrc 33712 \
|
|
--nwordstgt 33712 \
|
|
--joined-dictionary
|
|
|
|
sacrebleu -t wmt14/full -l de-en --echo src > $DATASET_DIR/sacrebleu_reference.de
|
|
|
|
cp $TEXT/code $DATASET_DIR/code
|