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
a397cebc
Commit
a397cebc
authored
Jul 12, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add back build_train_tower for compatibility
parent
3e1f0e53
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
0 deletions
+8
-0
tensorpack/train/feedfree.py
tensorpack/train/feedfree.py
+8
-0
No files found.
tensorpack/train/feedfree.py
View file @
a397cebc
...
...
@@ -6,6 +6,7 @@
import
tensorflow
as
tf
from
six.moves
import
zip
from
..utils
import
logger
from
..tfutils.gradproc
import
FilterNoneGrad
from
..tfutils.tower
import
TowerContext
,
get_current_tower_context
from
.input_source
import
QueueInput
,
FeedfreeInput
...
...
@@ -21,6 +22,13 @@ class FeedfreeTrainerBase(Trainer):
Expect ``self.data`` to be a :class:`FeedfreeInput`.
"""
# TODO deprecated
def
build_train_tower
(
self
):
logger
.
warn
(
"build_train_tower() was deprecated! Please build the graph "
"yourself, e.g. by self.model.build_graph(self._input_source)"
)
with
TowerContext
(
''
,
is_training
=
True
):
self
.
model
.
build_graph
(
self
.
_input_source
)
def
_setup
(
self
):
assert
isinstance
(
self
.
_input_source
,
FeedfreeInput
),
type
(
self
.
_input_source
)
self
.
_setup_input_source
(
self
.
_input_source
)
...
...
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