I fixed dim but still run problem

scripts/elfgames/tasks/start_client_mini.sh

@@ -13,7 +13,7 @@ echo $PYTHONPATH $SLURMD_NODENAME $CUDA_VISIBLE_DEVICES
 root=./myserver game=elfgames.tasks.game model=df_pred model_file=elfgames.tasks.df_model3 \
 stdbuf -o 0 -e 0 python -v ./selfplay.py \
     --T 1    --batchsize 10 \
-    --dim0 128    --dim1 128    --gpu 0 \
+    --dim0 2    --dim1 2    --gpu 0 \
     --keys_in_reply V rv    --mcts_alpha 0.03 \
     --mcts_epsilon 0.25    --mcts_persistent_tree \
     --mcts_puct 0.85    --mcts_rollout_per_thread 200 \

scripts/elfgames/tasks/start_server_mini.sh

@@ -29,7 +29,7 @@ save=./myserver game=elfgames.tasks.game model=df_kl model_file=elfgames.tasks.d
     --eval_winrate_thres 0.55     --port 1234 \
     --q_min_size 200     --q_max_size 4000 \
     --save_first     \
-    --num_block 20     --dim 128 \
+    --num_block 20     --dim 2 \
     --weight_decay 0.0002    --opt_method sgd \
     --bn_momentum=0 --num_cooldown=50 \
     --expected_num_client 1 \

src_py/elf/utils_elf.py

@@ -189,6 +189,7 @@ def copy_from(self, src):
         keys_extra = []
 
         for k, v in this_src.items():
+            print(k)
             # Copy it down to cpu.
             if k not in self.batch:
                 keys_extra.append(k)

src_py/elfgames/tasks/df_model3.py

@@ -158,8 +158,7 @@ def get_option_spec(cls):
         spec.addIntOption(
             'gpu',
             'which gpu to use',
-            -1)
-#            1)
+            1)
 
         spec.merge(GoResNet.get_option_spec())
 

src_py/rlpytorch/model_loader.py

@@ -50,7 +50,7 @@ def get_option_spec(cls, model_class=None, model_idx=None):
         spec.addIntOption(
             'gpu',
             'which GPU to use',
-            -1)
+            1)
         spec.addBoolOption(
             'check_loaded_options',
             'Toggles consistency check of loaded vs. current model options.',