Chris-Pedersen / LearnableWavelets

Learnable wavelet neural networks
5 stars 0 forks source link

Updated scattering modules to work with multiple input fields #33

Closed Chris-Pedersen closed 2 years ago

Chris-Pedersen commented 2 years ago

To run a comparison of our wavelet networks with the full 11-field CAMELs CNN from https://arxiv.org/pdf/2109.10360.pdf and https://arxiv.org/pdf/2109.10915.pdf

Chris-Pedersen commented 2 years ago

Hmm scattering networks are returning NaNs but not sure why, perhaps related to #21, will debug over the weekend

Chris-Pedersen commented 2 years ago

The NaNs only appear when sourcing my wavelet conda environment, without using conda it works fine so this PR is fine, and will continue with the NaN issue in #21