From 0f70f42495d6b580989004a71b66b09eee202dc9 Mon Sep 17 00:00:00 2001
From: tianzikang <tianzikang@CentOS7-Login2.future.cn>
Date: Fri, 8 Jul 2022 10:58:39 +0800
Subject: [PATCH] some change

---
 learn.py | 2 +-
 main.py  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/learn.py b/learn.py
index 1dd8c87..66604ab 100644
--- a/learn.py
+++ b/learn.py
@@ -29,7 +29,7 @@ def store_hyper_para(args, store_path):
     print('==================hyper parameters store done!==================')
     
 
-def dqn_learning(
+def qmix_learning(
     env_class,
     env_id,
     seed,
diff --git a/main.py b/main.py
index 80ad779..2b5febf 100644
--- a/main.py
+++ b/main.py
@@ -3,7 +3,7 @@
 from smac.env import StarCraft2Env
 
 from model import QMIX_agent
-from learn import dqn_learning
+from learn import qmix_learning
 from utils.schedule import LinearSchedule
 from utils.sc_wrapper import single_net_sc2env
 
@@ -50,7 +50,7 @@ def main(args=get_args()):
     else:
         env_class = StarCraft2Env
 
-    dqn_learning(
+    qmix_learning(
         env_class=env_class,
         env_id=args.map_name,
         seed=args.seed,