Remove unused variables from spatial transformer grad test

上级 f62a19d1
......@@ -2482,8 +2482,6 @@ def test_dnn_spatialtf_grad():
img = np.random.randint(low=0, high=256, size=img_dims)
# Convert from NHWC to NCHW
img = np.transpose(img, axes=(0, 3, 1, 2)).astype(theano.config.floatX)
scale_height = 0.25
scale_width = 0.75
# Transformation matrix
transform = [[-1, 0, 0],
......@@ -2503,7 +2501,7 @@ def test_dnn_spatialtf_grad():
grad = T.grad(None, wrt=[t_img, t_theta], known_grads={op: t_dy})
grad_fn = theano.function([t_img, t_theta, t_dy], grad)
dimg, dtheta = grad_fn(img, theta, dy)
grad_fn(img, theta, dy)
assert any([isinstance(node.op, dnn.GpuDnnTransformerGradI)
for node in grad_fn.maker.fgraph.toposort()])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论