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
797 stars 174 forks source link

🐛[BUG] CorrDiff: noise level inconsistent between regression model training and inference #447

Open nbren12 opened 2 months ago

nbren12 commented 2 months ago

Version

main

On which installation method(s) does this occur?

No response

Describe the issue

there appears to be an inconsistency in the sigma passed to the regression model at training and inference time. Training has random sigma. Inference has sigma = 1. Not sure if it would have any impact. I imagine that the regression model quickly learns to ignore the randomized sigma during training.

We should probably still fix it. Training should have a constant sigma.

Minimum reproducible example

No response

Relevant log output

No response

Environment details

No response

nbren12 commented 2 months ago

cc @mnabian @MortezaMardani