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

sparse tensor operations (#735)

Browse Source
This commit is contained in:
Martin Raison
2017-03-03 18:37:03 +01:00
committed by Adam Paszke
parent c93c884ee2
commit f17cfe4293
52 changed files with 2555 additions and 270 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/optim/optimizer.py
View File

@ -132,7 +132,8 @@ class Optimizer(object):
"""Clears the gradients of all optimized :class:`Variable` s."""
for group in self.param_groups:
for param in group['params']:
param.grad.data.zero_()
if param.grad is not None:
param.grad.data.zero_()
def step(self, closure):
"""Performs a single optimization step (parameter update).