Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
13723a91
Commit
13723a91
authored
Aug 09, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update docs
parent
2132599f
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
14 additions
and
7 deletions
+14
-7
.travis.yml
.travis.yml
+1
-7
tensorpack/train/base.py
tensorpack/train/base.py
+13
-0
No files found.
.travis.yml
View file @
13723a91
...
...
@@ -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
...
...
tensorpack/train/base.py
View file @
13723a91
...
...
@@ -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
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment