changed the config to do 4 layers instead of 2

SaltieRL · Jan 23, 2018 · eba36bd · eba36bd
1 parent 97bb8ed
commit eba36bd
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 7 deletions.
diff --git a/saltie2.cfg b/saltie2.cfg
@@ -26,8 +26,8 @@ model_name = TutorialModel
 control_scheme = regression_controls
 #model_directory = training/data/tutorial_bot_split10-layers/2/trained_variables
 teacher = replay_files
-num_layers = 3
-num_split_layers = 1
+num_layers = 4
+num_split_layers = 2
 num_width = 128
 is_graphing = True
 is_evaluating = True

diff --git a/trainer/configs/copy_trainer.cfg b/trainer/configs/copy_trainer.cfg
@@ -18,6 +18,7 @@ mini_batch_size = 5000
 num_width = 128
 model_package = models.actor_critic.tutorial_model
 model_name = TutorialModel
-num_layers = 3
-num_split_layers = 1
+num_layers = 4
+num_split_layers = 2
 keep_probability = 0.8
+should_regulate = False
diff --git a/trainer/configs/randomised_trainer.cfg b/trainer/configs/randomised_trainer.cfg
@@ -1,5 +1,5 @@
 [Randomised Trainer Configuration]
-total_batches = 2000
+total_batches = 4000
 save_step = 10
 teacher_package = TutorialBot.tutorial_bot_output
 #teacher_package = TutorialBot.atba2_demo_output
@@ -20,6 +20,7 @@ num_width = 128
 model_package = models.actor_critic.tutorial_model
 model_name = TutorialModel
 teacher = replay_files
-num_layers = 3
-num_split_layers = 1
+num_layers = 4
+num_split_layers = 2
 keep_probability = 0.8
+should_regulate = False