improve the training and validation datasets training plot in the train script

Edited by Ghost User