Visualising the filter parameter values and learning

[Chris-Pedersen]

Want to track this somehow, perhaps through wandb, but would be good to see just how much the filters change, where they converge to etc.

[Chris-Pedersen]

Have added some code to track wavelet parameters in wandb, will have to test if it slows down the training too much, I don't expect it should.

Still want to get a better handle on visualising the filters next.