search

Photon-AI-Research / NeuralSolvers

Neural network based solvers for partial differential equations and inverse problems :milky_way:. Implementation of physics-informed neural networks in pytorch.
MIT License
141 stars 46 forks source link

The for loops for pretraining and main training are ignored #36

Closed ahmedemam576 closed 3 years ago

ahmedemam576 commented 3 years ago

when choosing extremely small no. of samples , which lead that the len(no._of batches) of the dataset which is fed to the dataloader to be equal to zero, which makes the iteration loop to for i in range( 0,0, step): .. thats why the looping didn't work and the ic_loss wasn't calculated at the first place. i would suggest creating a warning message when choosing a no. of samples which can make the length of the datasets yields to zero after the using of "//" operator.

the for loops are found in line 503 and 518 in the pinn.fit () function in the pinn.py file

StillerPatrick commented 3 years ago

Will be fixed with pullrequest #40