Documentation

luoming3 · Jan 3, 2021 · 3bab66c · 3bab66c
1 parent 9411b4a
commit 3bab66c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/offpolicy/algorithms/mqmix/mqmix.py b/offpolicy/algorithms/mqmix/mqmix.py
@@ -11,6 +11,7 @@ class M_QMix:
     def __init__(self, args, num_agents, policies, policy_mapping_fn, device=torch.device("cuda:0"), vdn=False):
         """
         Trainer class for QMix with MLP policies. See parent class for more information.
+        :param vdn: (bool) whether the algorithm in  use is VDN.
         """
         self.args = args
         self.use_popart = self.args.use_popart