frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-31 04:04:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

chunky sync again

Browse Source
This commit is contained in:
Yangqing Jia
2016-08-01 20:58:46 -07:00
parent a7af924919
commit c15e45c9bb
47 changed files with 2065 additions and 278 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
caffe2/python/gradient_checker.py
View File

@ -106,7 +106,11 @@ class GradientChecker:
outputs_with_grads
)
grad_estimate = np.zeros_like(inputs[input_to_check])
assert grad_estimate.shape == grad.shape, input_to_check
if grad_estimate.shape != grad.shape:
raise Exception(
"Mismatched gradient shapes: estimated ({}), grad ({})".format(
grad_estimate.shape, grad.shape))
for current_dim in range(dims_to_check):
# Positive gradient
inputs[input_to_check].flat[current_dim] += self._stepsize