Relative Content

Tag Archive for pytorchneural-network

Implementing GradNorm for a Physics Informed Neural Network in pytorch

I am currently writing a physics informed neural network that learns the velocity, temperature, and pressure field from the evolution of a bubble in a pool boiling scenario. I have tried to implement the GradNorm regularization scheme to help convergence of the model as outlined here https://arxiv.org/abs/2308.08468 , but instead of helping convergence it prevents it.