PlanarLayer parameter b is not Tracked

[turinglang]

PlanarLayer parameter b is not Tracked

[torfjelde]

Ah crap, didn't think of that. It's because Flux.params filters out non-arrays.

A simple solution is to just to keep track of the parameters outside of the model, i.e. not using Flux.params.
Alternatively, you can wait for #123 👍

[torfjelde]

Ah, thanks for pointing this out! This was introduced not too long ago in a PR and the change sort of slipped under the radar.

For the moment, I'd manually instantiate PlanarLayer. Under the hood it's literally just doing PlanarLayer(randn(2), randn(2), randn()) and it was supposed to wrap the arguments in param.

[torfjelde]

Fix is one the way!

[cmerkatas]

Thank you @torfjelde. It will be useful to fit flows I believe. I haven't managed to "track" it even manually but it was a quick test. I will try again later.

[torfjelde]

Should be fixed now, i.e. you can do PlanarLayer(2, Tracker.param) 👍

[torfjelde]

Fixed in #122

[cmerkatas]

Thank you, that fix really does Track the components. However, there is still some problem with the examples. For example

using Bijectors
using Flux, Tracker

d = MvNormal(zeros(2), ones(2));
b = PlanarLayer(2, Tracker.param);
flow = transformed(d,b)
# after this point b is not tracked anymore 

@Flux.treelike Composed
@Flux.treelike Bijectors.TransformedDistribution
@Flux.treelike PlanarLayer
ps = Flux.params(flow)

Params([[-0.5236266422759951, -0.05265731672851134] (tracked), [1.1964638246267423, -0.5138318038308557] (tracked)])

Am I missing something?

[cmerkatas]

Ah got it. Thanks for your reply!