|
Algorithm 2 CLI-PPO End-to-End Training Framework |
-
1:
Initialize PPO policy parameters , LSTM parameters , ICM parameters , rollout buffer and success buffer
-
2:
Set , ,
-
3:
Set learning rate , discount factor , intrinsic reward coefficient , and curriculum threshold
-
4:
Initialize curriculum boundary ratio
-
5:
for to do
-
6:
for to do
-
7:
for to N do
-
8:
Obtain observation:
-
9:
Update recurrent hidden states:
-
10:
Sample continuous action:
-
11:
Execute action:
-
12:
Compute intrinsic curiosity reward:
-
13:
Compute total reward:
-
14:
Store transition:
-
15:
end for
-
16:
end for
-
17:
Compute ICM loss:
-
18:
Update ICM parameters:
-
19:
Update PPO policy:
-
20:
Update success buffer:
-
21:
if then
-
22:
Expand curriculum boundary:
-
23:
Update environment boundary:
-
24:
end if
-
25:
Clear rollout buffer:
-
26:
end for
-
27:
return
|