search

ebetica / autogradpp

Direct C++ Interface to PyTorch
MIT License
80 stars 12 forks source link

SGD: not launch a CUDA kernel if weight_decay_ = 0 #30

Closed eugene-kharitonov closed 6 years ago

eugene-kharitonov commented 6 years ago

just as Adam