提交 1e3de2ce authored 作者: Arjun Jain's avatar Arjun Jain

Changes suggested by Fred

上级 74ea01ac
......@@ -76,7 +76,7 @@ void im2col(const float* data_im, const int channels,
// Author: Arjun Jain
CudaNdarray* validMM(const CudaNdarray *input,
CudaNdarray* corrMM(const CudaNdarray *input,
CudaNdarray *weight,
CudaNdarray *output,
int padding = 0)
......
......@@ -830,6 +830,8 @@ def test_gemm():
input: (batch size, channels, rows, columns)
filters: (number of filters, channels, rows, columns)
"""
for mode in ['valid', 'full']:
print 'Testing mode: ' + mode
for bs in range(1, 5):
for ch in range(1,4):
for nf in range(1,4):
......@@ -839,7 +841,6 @@ def test_gemm():
kshape = (nf, ch, rFlt, rFlt)
print "ishape: ", ishape
print "kshape: ", kshape
mode = 'full'
subsample = (1, 1)
npy_img = theano._asarray(numpy.random.rand(*ishape), dtype='float32')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论