some change

Felixvillas · Jul 8, 2022 · 0f70f42 · 0f70f42
1 parent 68a5cc6
commit 0f70f42
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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,