I fixed a few small conversion problems, mostly related to wandb logging and the PPO version. I also added a few more print statements and comments for where we can improve speed. One speed improvement was to use the slicing that is already used in dynamics.calculate_loss() in dynamics.predict_features(). I merged these two functions and call it predict_features for clarity everywhere now.
I fixed a few small conversion problems, mostly related to wandb logging and the PPO version. I also added a few more print statements and comments for where we can improve speed. One speed improvement was to use the slicing that is already used in dynamics.calculate_loss() in dynamics.predict_features(). I merged these two functions and call it predict_features for clarity everywhere now.