提交 14bdbaa5 authored 作者: notoraptor's avatar notoraptor

Update test cases for runtitme algorithms.

Fix typos.
上级 a346913f
...@@ -9,10 +9,10 @@ import sys ...@@ -9,10 +9,10 @@ import sys
import theano import theano
from theano.configdefaults import SUPPORTED_DNN_CONV_ALGO_RUNTIME from theano.configdefaults import SUPPORTED_DNN_CONV_ALGO_RUNTIME
from theano.tensor.nnet.abstract_conv import get_conv_output_shape
from theano.gpuarray.cudnn_defs import (HALF, FLOAT, DOUBLE, from theano.gpuarray.cudnn_defs import (HALF, FLOAT, DOUBLE,
TRUE_HALF_CONFIG, PSEUDO_HALF_CONFIG, FLOAT_CONFIG, DOUBLE_CONFIG) TRUE_HALF_CONFIG, PSEUDO_HALF_CONFIG, FLOAT_CONFIG, DOUBLE_CONFIG)
from theano.gpuarray.tests.check_dnn_conv import (cudnn, TestDnnConv2D, TestDnnConv3D, CheckDnn) from theano.gpuarray.tests.check_dnn_conv import (cudnn, TestDnnConv2D, TestDnnConv3D, CheckDnn)
from theano.tensor.nnet.abstract_conv import get_conv_output_shape
class TupleAction(argparse.Action): class TupleAction(argparse.Action):
...@@ -32,7 +32,7 @@ class BorderAction(TupleAction): ...@@ -32,7 +32,7 @@ class BorderAction(TupleAction):
if __name__ != '__main__': if __name__ != '__main__':
raise ImportError('This script cannot be imported.') raise ImportError('This script must not be imported.')
args = sys.argv[1:] args = sys.argv[1:]
computations = FWD, BWD_FILTER, BWD_DATA = ('fwd', 'gradweight', 'gradinput') computations = FWD, BWD_FILTER, BWD_DATA = ('fwd', 'gradweight', 'gradinput')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论