search

jcreinhold / synthtorch

deep neural network-based image translation/synthesis
Other
27 stars 7 forks source link

Dropout not working in fa-train #1

Closed jcreinhold closed 5 years ago

jcreinhold commented 5 years ago

At least when 1-cycle policy is enabled, dropout > 0 causes backprop to fail. It states that an inplace operation has changed a value needed for the gradient.

jcreinhold commented 5 years ago

Set inplace=False in dropout blocks solved the issue