Remove ReduceLRUponNan from callbacks
Remove ReduceLRUponNan from callbacks

Also stop training model if nan or inf loss

materialsvirtuallab/megnet,materialsvirtuallab/megnet,materialsvirtuallab/megnet,materialsvirtuallab/megnet,materialsvirtuallab/megnet