search

m2lines / gz21_ocean_momentum

Stochastic-Deep Learning Parameterization of Ocean Momentum Forcing
MIT License
5 stars 1 forks source link

PyTorch runtime error during training: `input type (double) and bias type (float) should be the same` #117

Open raehik opened 8 months ago

raehik commented 8 months ago

Sometimes while training get a PyTorch Runtime Error: input type (double) and bias type (float) should be the same. It seems random. I can't tell what it's triggered by. Model parameter and training data types seem to match just fine -- I don't know where the double is coming from.