diff --git a/scripts/elfgames/tasks/start_client_mini.sh b/scripts/elfgames/tasks/start_client_mini.sh
index f46ec62..9bfbf1a 100755
--- a/scripts/elfgames/tasks/start_client_mini.sh
+++ b/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 \
diff --git a/scripts/elfgames/tasks/start_server_mini.sh b/scripts/elfgames/tasks/start_server_mini.sh
index c224a80..654ab6b 100755
--- a/scripts/elfgames/tasks/start_server_mini.sh
+++ b/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 \
diff --git a/src_py/elf/utils_elf.py b/src_py/elf/utils_elf.py
index db5c3b2..8d5c7f3 100644
--- a/src_py/elf/utils_elf.py
+++ b/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)
diff --git a/src_py/elfgames/tasks/df_model3.py b/src_py/elfgames/tasks/df_model3.py
index 8ef1793..4c9ccb0 100644
--- a/src_py/elfgames/tasks/df_model3.py
+++ b/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())
 
diff --git a/src_py/rlpytorch/model_loader.py b/src_py/rlpytorch/model_loader.py
index 5e50205..342258c 100644
--- a/src_py/rlpytorch/model_loader.py
+++ b/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.',