提交 4a13ecd7 编辑于 作者: Haowen Xu's avatar Haowen Xu
浏览文件

follow upstream versions

上级 4fde5e29

.travis.yml

+2 −2
原始行号 差异行号 差异行
@@ -6,8 +6,8 @@ env: 
  matrix:

  - PYTHON_VERSION=2 TENSORFLOW_VERSION=1.5

  - PYTHON_VERSION=3 TENSORFLOW_VERSION=1.5

  - PYTHON_VERSION=2 TENSORFLOW_VERSION=1.6

  - PYTHON_VERSION=3 TENSORFLOW_VERSION=1.6

  - PYTHON_VERSION=2 TENSORFLOW_VERSION=1.8

  - PYTHON_VERSION=3 TENSORFLOW_VERSION=1.8

install:

  - docker pull "ipwx/travis-tensorflow-docker:py${PYTHON_VERSION}tf${TENSORFLOW_VERSION}"

script:

donut/training.py

+3 −3
原始行号 差异行号 差异行 
import six

import numpy as np

import tensorflow as tf

from tfsnippet.scaffold import train_loop, TrainLoopContext

from tfsnippet.scaffold import train_loop, TrainLoop

from tfsnippet.utils import (VarScopeObject,

                             reopen_variable_scope,

                             get_default_session_or_error,
@@ -257,12 +257,12 @@ class DonutTrainer(VarScopeObject): 


        # training loop

        lr = self._initial_lr

        with train_loop(

        with TrainLoop(

                param_vars=self._train_params,

                early_stopping=True,

                summary_dir=summary_dir,

                max_epoch=self._max_epoch,

                max_step=self._max_step) as loop:  # type: TrainLoopContext

                max_step=self._max_step) as loop:  # type: TrainLoop

            loop.print_training_summary()



            for epoch in loop.iter_epochs():