Deploying the pretrained denoiser models in user code requires a manual copy of the InitialEmbedding from the demo notebook.
Solution
Adding the InitialEmbedding to the graphite module solves this issue.
Open questions
I placed the InitialEmbedding in nn/utils/e3nn_initial_embedding trying to mimic the directory/file structure found in nn/models and nn/conv.
Given your future plans for this project a different file might be more appropriate.
Issue
Deploying the pretrained denoiser models in user code requires a manual copy of the InitialEmbedding from the demo notebook.
Solution
Adding the InitialEmbedding to the graphite module solves this issue.
Open questions
I placed the InitialEmbedding in nn/utils/e3nn_initial_embedding trying to mimic the directory/file structure found in nn/models and nn/conv. Given your future plans for this project a different file might be more appropriate.