Skip to content

Commit 4968c77

Browse files
Minor
1 parent bc1e53f commit 4968c77

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/ppo_with_pytorch.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -349,12 +349,12 @@ def progress(num_steps, metrics):
349349
eval_sps.append(metrics['speed/eval_sps'])
350350
train_sps.append(metrics['speed/sps'])
351351

352-
plt.xlim([0, 30_000_000])
353-
plt.ylim([0, 6000])
354-
plt.xlabel('# environment steps')
355-
plt.ylabel('reward per episode')
356-
plt.plot(xdata, ydata)
357-
plt.show()
352+
# plt.xlim([0, 30_000_000])
353+
# plt.ylim([0, 6_000])
354+
# plt.xlabel('# environment steps')
355+
# plt.ylabel('reward per episode')
356+
# plt.plot(xdata, ydata)
357+
# plt.show()
358358

359359
agent, env = train(progress_fn=progress)
360360

0 commit comments

Comments
 (0)