Fixed a bug in affine_. The layer was implemented as an inplace layer,
however, that doesn't really work if we need to compute the gradients with respect to the parameters which the layer attempted to do. So I made the layer's parameters non-learnable so it can continue to be inplace since the main usecase for this layer is to be a fast replacement for bn_ during testing.
Showing
Please
register
or
sign in
to comment