Skip to content

Commit fe91ee4

Browse files
committed
upload curiosity model
1 parent eaf04ac commit fe91ee4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

contents/Curiosity_Model/Curiosity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def learn(self):
129129
bs, ba, br, bs_ = batch_memory[:, :self.n_s], batch_memory[:, self.n_s], \
130130
batch_memory[:, self.n_s + 1], batch_memory[:, -self.n_s:]
131131
self.sess.run(self.dqn_train, feed_dict={self.tfs: bs, self.tfa: ba, self.tfr: br, self.tfs_: bs_})
132-
if self.learn_step_counter % 1000 == 0:
132+
if self.learn_step_counter % 1000 == 0: # delay training in order to stay curious
133133
self.sess.run(self.dyn_train, feed_dict={self.tfs: bs, self.tfa: ba, self.tfs_: bs_})
134134
self.learn_step_counter += 1
135135

0 commit comments

Comments
 (0)