wassname/rl_2d_walker.js
1
0
Fork 0
mirror of https://github.com/wassname/rl_2d_walker.js.git synced 2026-09-09 11:33:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix env interface

Browse Source
This commit is contained in:
wassname
2018-12-01 17:14:20 +08:00
parent 4be259e31a
commit d296c91508
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/js/ddpg/ddpg_agent.js
+1 -3
View File
@@ -142,12 +142,10 @@ class DDPGAgent {
const tfActions = this.ddpg.perturbedPrediction(tfPreviousStep);
// Step in the environment with theses actions
let mAcions = tfActions.buffer().values;
let mReward = this.env.step(mAcions);
let [mState, mReward, mDone, info] = this.env.step(mAcions);
this.rewardsList.push(mReward);
// Get the new observations
let mState = this.env.getState();
let tfState = tf.tensor2d([mState]);
let mDone = 0;
if (mReward == -1 && this.config.stopOnRewardError){
mDone = 1;
}