Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Auto merge of #87 - alexandermorozov:fix-sgd-init, r=hobofan
fix/sgd: initialize weight gradient history with zeroes SGD solver used unintialized history tensors. If there were some NaNs then whole network got poisoned after the first generation even if momentum was set to zero. This patch prefills gradient history with zeros. FIX: autumnai/leaf-examples#13
- Loading branch information