PPO playing MountainCar-v0 from https://github.com/sgoodfriend/rl-algo-impls/tree/5598ebc4b03054f16eebe76792486ba7bcacfc5c
68e589c
from abc import ABC, abstractmethod | |
class Callback(ABC): | |
def __init__(self) -> None: | |
super().__init__() | |
self.timesteps_elapsed = 0 | |
def on_step(self, timesteps_elapsed: int = 1) -> bool: | |
self.timesteps_elapsed += timesteps_elapsed | |
return True |