added switch -t or -v in the CLI to choose between training or
validation