fixed bug

George-Chia · Apr 12, 2021 · 80829b9 · 80829b9
1 parent 60d6e76
commit 80829b9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/magic.py b/magic.py
@@ -112,7 +112,7 @@ def forward(self, x, info={}):
         else:
             adj1 = self.get_complete_graph(agent_mask)
 
-		comm = F.elu(self.gconv1(comm, adj1))
+        comm = F.elu(self.gconv1(comm, adj1))
 
         if self.args.learn_second_graph and not self.args.second_graph_complete:
             if self.args.use_gconv_encoder:

diff --git a/train_grf.sh b/train_grf.sh
@@ -5,7 +5,7 @@ python -u main.py \
   --env_name grf \
   --nagents 3 \
   --nprocesses 16 \
-  --num_epochs 500 \
+  --num_epochs 300 \
   --epoch_size 10 \
   --hid_size 128 \
   --detach_gap 10 \
@@ -32,8 +32,8 @@ python -u main.py \
   --reward_type scoring \
   --save \
   --save_every 200 \
-  --seed 0 \
+  --seed 5678 \
   --plot \
-  --plot_env magic_grf_seed_0 \
-  --plot_port 8097 \
+  --plot_env magic_grf_seed_5678 \
+  --plot_port 8009 \
   | tee train_grf.log