Commit 13723a91 authored by Yuxin Wu's avatar Yuxin Wu

update docs

parent 2132599f
......@@ -41,10 +41,9 @@ matrix:
install:
- pip install -U pip # the pip version on travis is too old
- pip install flake8 scikit-image opencv-python lmdb h5py msgpack
- pip install .
- pip install flake8 scikit-image opencv-python lmdb h5py msgpack
# check that dataflow can be imported alone
- python -c "import pyarrow"
- python -c "import tensorpack.dataflow"
- ./tests/install-tensorflow.sh
......@@ -66,11 +65,6 @@ script:
- cd $TRAVIS_BUILD_DIR # go back to root so that deploy may work
notifications:
email:
recipients:
- ppwwyyxxc@gmail.com
on_success: never
on_failure: always
webhooks:
urls:
- https://webhooks.gitter.im/e/cede9dbbf6630b3704b3
......
......@@ -100,6 +100,19 @@ class Trainer(object):
Certain callbacks will only be run by chief worker.
"""
sess = None
"""
The ``tf.Session`` object the trainer is using.
Available after :meth:`initialize()`.
"""
hooked_sess = None
"""
The ``tf.train.MonitoredSession`` object the trainer is using.
It contains all the hooks the callbacks have registered.
Available after :meth:`initialize()`.
"""
def __init__(self):
self._callbacks = []
self.loop = TrainLoop()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment