train_bootstrapped.sh 2.9 KB

123456789101112131415161718192021222324252627282930313233343536
  1. PYTHON=$(which python)
  2. # python -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type easy -ng 25 -att -student -fr -th 0.5 -prune"
  3. if [ $1 = "bootstrap_easy" ]; then
  4. screen -S "easy_25" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type easy -ng 25 -att -student -fr -th 0.5 -prune"
  5. screen -S "easy_50" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type easy -ng 50 -att -student -fr -th 0.5 -prune"
  6. screen -S "easy_500" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type easy -ng 500 -att -student -fr -th 0.5 -prune"
  7. fi
  8. if [ $1 = "bootstrap_medium" ]; then
  9. screen -S "medium_50" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type medium -ng 50 -att -student -fr -th 0.7 -prune"
  10. screen -S "medium_100" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type medium -ng 100 -att -student -fr -th 0.7 -prune"
  11. screen -S "medium_500" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type medium -ng 500 -att -student -fr -th 0.7 -prune"
  12. fi
  13. if [ $1 = "bootstrap_hard" ]; then
  14. screen -S "hard_50" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type hard -ng 50 -att -student -fr -th 0.7 -prune"
  15. screen -S "hard_100" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_policy_qlearn -c config -type hard -ng 100 -att -student -fr -th 0.7 -prune"
  16. fi
  17. if [ $1 = "base_drqn_easy" ]; then
  18. screen -S "base_easy_25" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type easy -ng 25 -att -fr"
  19. screen -S "base_easy_50" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type easy -ng 50 -att -fr "
  20. screen -S "base_easy_500" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type easy -ng 500 -att -fr"
  21. fi
  22. if [ $1 = "base_drqn_medium" ]; then
  23. screen -S "base_medium_50" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type medium -ng 50 -att -fr"
  24. screen -S "base_medium_100" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type medium -ng 100 -att -fr"
  25. screen -S "base_medium_500" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type medium -ng 500 -att -fr"
  26. fi
  27. if [ $1 = "base_drqn_hard" ]; then
  28. screen -S "base_hard_50" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type hard -ng 50 -att -fr"
  29. screen -S "base_hard_100" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type hard -ng 100 -att -fr"
  30. # screen -S "base_hard_500" -dm bash -c "${PYTHON} -m crest.agents.lstm_drqn.train_single_generate_agent -c config -type hard -ng 500 -att -fr"
  31. fi