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
fe41b5f0
Commit
fe41b5f0
authored
Apr 15, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
remove a test; fix names in DataParallelOfflinePredictor (fix #730)
parent
fb5a99e0
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
12 additions
and
10 deletions
+12
-10
tensorpack/predict/multigpu.py
tensorpack/predict/multigpu.py
+1
-1
tests/test_resnet.py
tests/test_resnet.py
+11
-9
No files found.
tensorpack/predict/multigpu.py
View file @
fe41b5f0
...
@@ -95,7 +95,7 @@ class DataParallelOfflinePredictor(OnlinePredictor):
...
@@ -95,7 +95,7 @@ class DataParallelOfflinePredictor(OnlinePredictor):
for
idx
,
t
in
enumerate
(
towers
):
for
idx
,
t
in
enumerate
(
towers
):
tower_name
=
'tower'
+
str
(
t
)
tower_name
=
'tower'
+
str
(
t
)
inputs_desc
=
[
InputDesc
(
desc
.
type
,
desc
.
shape
,
tower_name
+
'
/
'
+
desc
.
name
)
inputs_desc
=
[
InputDesc
(
desc
.
type
,
desc
.
shape
,
tower_name
+
'
_
'
+
desc
.
name
)
for
desc
in
config
.
inputs_desc
]
for
desc
in
config
.
inputs_desc
]
input
=
PlaceholderInput
()
input
=
PlaceholderInput
()
input
.
setup
(
inputs_desc
)
input
.
setup
(
inputs_desc
)
...
...
tests/test_resnet.py
View file @
fe41b5f0
from
case_script
import
TestPythonScript
from
case_script
import
TestPythonScript
# noqa
# this tests occasionally fails (memory issue on travis?)
class
ResnetTest
(
TestPythonScript
):
@
property
def
script
(
self
):
return
'../examples/ResNet/imagenet-resnet.py'
def
test
(
self
):
# class ResnetTest(TestPythonScript):
self
.
assertSurvive
(
# @property
self
.
script
,
# def script(self):
args
=
[
'--fake'
,
'--data_format NHWC'
],
timeout
=
20
)
# return '../examples/ResNet/imagenet-resnet.py'
#
# def test(self):
# self.assertSurvive(
# self.script,
# args=['--fake', '--data_format NHWC'], timeout=20)
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