Training fails after 70 iterations if using self.initialize_NN(self.uv_layers)

[pengguo-seismo]

Hi Chengping,

Thank you for sharing a very interesting paper and the code!

I have a issue when running the code, and hope that you may help me out.

The problem is that, if I didn't use the existed file "uv_NN_10s.pickle" to initialize the network, and instead using the function initialize_NN to intialize the weights and biases, the training will fail after 70 iterations. Any insights for how to properly intialize the nn?

Thank yo so much in advance.

Regards