Jhin4 commited on
Commit
249fd76
1 Parent(s): 48c662e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -2
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
+