Update README.md
Browse files
README.md
CHANGED
@@ -33,5 +33,25 @@ TODO: Add your code
|
|
33 |
from stable_baselines3 import ...
|
34 |
from huggingface_sb3 import load_from_hub
|
35 |
|
36 |
-
|
37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
from stable_baselines3 import ...
|
34 |
from huggingface_sb3 import load_from_hub
|
35 |
|
36 |
+
#The hyper-parameter
|
37 |
+
model = DQN(
|
38 |
+
"MlpPolicy",
|
39 |
+
env=env,
|
40 |
+
buffer_size=100000,
|
41 |
+
learning_starts=50000,
|
42 |
+
batch_size=128,
|
43 |
+
train_freq=1,
|
44 |
+
gradient_steps=3,
|
45 |
+
tau=1.0,
|
46 |
+
gamma=0.99,
|
47 |
+
learning_rate=0.0001,
|
48 |
+
target_update_interval=10000,
|
49 |
+
exploration_initial_eps=1.0,
|
50 |
+
exploration_fraction=0.1,
|
51 |
+
exploration_final_eps=0.05,
|
52 |
+
policy_kwargs=dict(net_arch=[256, 256]),
|
53 |
+
device='auto',
|
54 |
+
verbose=1
|
55 |
+
)
|
56 |
+
|
57 |
+
|