提交 a65b0a11 authored 作者: affanv14's avatar affanv14

fix variable dtype typo

上级 f3f8da72
...@@ -1799,7 +1799,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer): ...@@ -1799,7 +1799,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
if type(node.op) in [AbstractConv2d, AbstractConv3d]: if type(node.op) in [AbstractConv2d, AbstractConv3d]:
img, kern = node.inputs img, kern = node.inputs
for(var, shape) in zip((img, kern), shapes): for(var, shape) in zip((img, kern), shapes):
result[var] = theano.shared(np.random.random(shape).astype(theano.config.floatX), result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
var.name, borrow=True) var.name, borrow=True)
if type(node.op) in [AbstractConv2d_gradWeights, AbstractConv3d_gradWeights]: if type(node.op) in [AbstractConv2d_gradWeights, AbstractConv3d_gradWeights]:
...@@ -1814,7 +1814,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer): ...@@ -1814,7 +1814,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
result[kshape] = theano.tensor.as_tensor_variable(node.op.kshp[-2:]) result[kshape] = theano.tensor.as_tensor_variable(node.op.kshp[-2:])
for(var, shape) in zip((img, top), (node.op.imshp, tshp)): for(var, shape) in zip((img, top), (node.op.imshp, tshp)):
result[var] = theano.shared(np.random.random(shape).astype(theano.config.floatX), result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
var.name, borrow=True) var.name, borrow=True)
if type(node.op) in [AbstractConv2d_gradInputs, AbstractConv3d_gradInputs]: if type(node.op) in [AbstractConv2d_gradInputs, AbstractConv3d_gradInputs]:
...@@ -1829,7 +1829,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer): ...@@ -1829,7 +1829,7 @@ class ConvMetaOptimizer(LocalMetaOptimizer):
result[ishape] = theano.tensor.as_tensor_variable(node.op.imshp[-2:]) result[ishape] = theano.tensor.as_tensor_variable(node.op.imshp[-2:])
for(var, shape) in zip((kern, top), (node.op.kshp, tshp)): for(var, shape) in zip((kern, top), (node.op.kshp, tshp)):
result[var] = theano.shared(np.random.random(shape).astype(theano.config.floatX), result[var] = theano.shared(np.random.random(shape).astype(var.dtype),
var.name, borrow=True) var.name, borrow=True)
return result return result
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论