train_neumann_grad_scoring.py change name to train_neumann.py ?
train_neumann_grad_scoring.py change name to train_neumann.py ?