fix Q-leaning

2021-02-27 16:58:40 +08:00
parent fd84a59875
commit 23a9cf2cba
1 changed files with 0 additions and 5 deletions
--- a/codes/Q-learning/main.py
+++ b/codes/Q-learning/main.py
@@ -136,11 +136,6 @@ def eval(cfg, saved_model_path = SAVED_MODEL_PATH):
    plot(rewards)
    plot(MA_rewards,ylabel='moving_average_rewards_train')
      
-def main():
-    cfg = get_args()
-    # train(cfg)
-    test(cfg)
-
 if __name__ == "__main__":
    cfg = get_args()
    if cfg.train: