sgoodfriend's picture
PPO playing QbertNoFrameskip-v4 from https://github.com/sgoodfriend/rl-algo-impls/tree/2067e21d62fff5db60168687e7d9e89019a8bfc0
5c87b65
raw
history blame
269 Bytes
from abc import ABC
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