Crop Image / Training now works / Got memory error while training

This commit is contained in:
Tobi
2019-05-22 16:50:35 +02:00
parent 82a4c23258
commit 8082fde44c
9 changed files with 58244 additions and 40 deletions
+22
View File
@@ -0,0 +1,22 @@
default:
trainer: ppo
batch_size: 1024
beta: 5.0e-3
buffer_size: 10240
epsilon: 0.2
gamma: 0.99
hidden_units: 128
lambd: 0.95
learning_rate: 3.0e-4
max_steps: 5.0e4
memory_size: 256
normalize: false
num_epoch: 3
num_layers: 2
time_horizon: 64
sequence_length: 64
summary_freq: 1000
use_recurrent: false
use_curiosity: false
curiosity_strength: 0.01
curiosity_enc_size: 128