NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
https://developer.nvidia.com/modulus
Apache License 2.0
937 stars 219 forks source link

Add missing parameter to healpix convolutional layers #563

Closed daviddpruitt closed 3 months ago

daviddpruitt commented 3 months ago

Add n_layers parameter in initialization of helapix convolutional layers.

Modulus Pull Request

Description

Add n_layers parameter in initialization of helapix convolutional layers. For hydra instantiation support all healpix convolutional layers need the n_layers parameter. The parameter was missing from some layers causing a crash.

Checklist

Dependencies

mnabian commented 3 months ago

/blossom-ci

mnabian commented 3 months ago

/blossom-ci

mnabian commented 3 months ago

/blossom-ci