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):
for idx, t in enumerate(towers):
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]
input = PlaceholderInput()
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):
self.assertSurvive(
self.script,
args=['--fake', '--data_format NHWC'], timeout=20)
# class ResnetTest(TestPythonScript):
# @property
# def script(self):
# 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