If would be nice to be able to specify the validation dataset as an argument of train_net.py and test_net.py respectively.