evaluation_script_zero_shot.sh 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. # 1- gametype
  2. # 2- numgames
  3. # dqn no att
  4. PYTHON=$(which python)
  5. # # drqn att
  6. ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -level 15 -att -level 20
  7. if [ $1 = "easy" ]; then
  8. # drqn att
  9. ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -att -student -th 0.5 -prune -level 20
  10. fi
  11. if [ $1 = "medium" ]; then
  12. ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -att -student -th 0.7 -prune -level 20
  13. fi
  14. if [ $1 = "hard" ]; then
  15. ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -att -student -th 0.7 -prune -level 20
  16. fi
  17. # drqn att
  18. # ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -level 15 -att -level 25
  19. if [ $1 = "easy" ]; then
  20. # drqn att
  21. ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -att -student -th 0.5 -prune -level 25
  22. fi
  23. if [ $1 = "medium" ]; then
  24. ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -att -student -th 0.7 -prune -level 25
  25. fi
  26. if [ $1 = "hard" ]; then
  27. ${PYTHON} -m textrl.agents.lstm_drqn.evaluate_agents_att -c config -type ${1} -ng ${2} -att -student -th 0.7 -prune -level 25
  28. fi