https://github.com/tensorflow/tensor2tensor
Raw File
Tip revision: 0d840eeadcd33b31e07b650308788a121b85ddb1 authored by Lukasz Kaiser on 21 March 2019, 23:02:05 UTC
Make TransformerLM train reasonably well in trax. Adding loss and metric masking and dropout refactor in Transformer.
Tip revision: 0d840ee
.travis.yml
sudo: required
language: python
cache: pip
git:
  depth: 3
  quiet: true
services:
  - docker
python:
  - "2.7"
  - "3.6"
env:
  global:
    - T2T_PROBLEM=algorithmic_reverse_binary40_test
    - T2T_DATA_DIR=/tmp/t2t-data
    - T2T_TRAIN_DIR=/tmp/t2t-train
    - TF_LATEST="1.13.*"
    # This is necessary to have gsutil work with Python 2.7
    - BOTO_CONFIG=/dev/null
  matrix:
    # We test against the latest stable TensorFlow and tf-nightly.
    # If updating, also update TF_LATEST above
    - TF_VERSION="1.13.*"
    - TF_VERSION="tf-nightly"
install:
  - ./oss_scripts/oss_pip_install.sh
script:
  - ./oss_scripts/oss_tests.sh
  - ./oss_scripts/oss_integration_test.sh

  # Conditional commands should each be in a separate block to get proper
  # errors on Travis.
  #
  # TODO(afrozm): Re-enable if this becomes an issue.
  # - if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]] && [[ "$TF_VERSION" == "tf-nightly"  ]]; then
  #       pylint -j 2 tensor2tensor;
  #   fi
back to top