search

FluxML / Torch.jl

Sensible extensions for exposing torch in Julia.
Other
213 stars 15 forks source link

Make MaxPool grads work #13

Closed DhairyaLGandhi closed 4 years ago

DhairyaLGandhi commented 4 years ago

Probably good to say here that we have added way more than the maxpool grads. It now adds most gradients needed to get ResNet train with Zygote. We are getting close to getting this merged, with tests, docs, etc.

DhairyaLGandhi commented 4 years ago

Locally this allows for taking gradients through a ResNet using Zygote

DhairyaLGandhi commented 4 years ago

bors ping