update train script

ctcvr_train.sh

-if ! ps aux | grep src/train_diary.py | grep -v grep
+if [ -z $1 ];
 then
-  nohup /home/gmuser/.virtualenvs/tf1/bin/python3 /srv/apps/gm_strategy_cvr/src/train_diary.py > ~/ctcvr_diary.log &
-fi
-
-if ! ps aux | grep src/train_tractate.py | grep -v grep
+  echo "You should input one parameter"
+elif [ $1 == "diary" ] || [ $1 == "tractate" ];
 then
-  nohup /home/gmuser/.virtualenvs/tf1/bin/python3 /srv/apps/gm_strategy_cvr/src/train_tractate.py > ~/ctcvr_tractate.log &
+  echo $1;
+  nohup /home/gmuser/.virtualenvs/tf1/bin/python3 /srv/apps/gm_strategy_cvr/src/train_$1.py > ~/ctcvr_$1.log &
 fi