Commit fe41b5f0 authored by Yuxin Wu's avatar Yuxin Wu

remove a test; fix names in DataParallelOfflinePredictor (fix #730)

parent fb5a99e0
...@@ -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)
......
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)
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