fix iteration increments

leggedrobotics / rsl_rl

Fast and simple implementation of RL algorithms, designed to run fully on GPU.

Other

505 stars 156 forks source link

fix iteration increments #8

Closed sheim closed 10 months ago

sheim commented 2 years ago

In on_policy_learning.py, current_learning_iteration is only being incremented at the end of the learning run, so all intermediate saves have the incorrect iteration. This is fixed, so if you resume with an intermediate checkpoint, it starts counting from the correct place (mainly just a nice-to-have for the tensorboard visualizations).

Mayankm96 commented 10 months ago

Closing this for now. Please reopen if the issue is still present on master. Thanks a lot!