Chris-Pedersen / LearnableWavelets

Learnable wavelet neural networks
5 stars 0 forks source link

Visualising learned filters #24

Closed Chris-Pedersen closed 2 years ago

Chris-Pedersen commented 2 years ago

The filter params are stored and easily accessible via a wandb api, which can take care of all our bookkeeping for various training runs. So it would be nice to have a script to take a wandb run, and for a chosen epoch plot visualisations of the filters. We could also experiment with storing these visualisations at certain intervals automatically using wandb, as long as it doesn't get too bulky.

Chris-Pedersen commented 2 years ago

Notebook to do this in #23