How to use sternn/atari_unit3 with stable-baselines3:
from huggingface_sb3 import load_from_hub checkpoint = load_from_hub( repo_id="sternn/atari_unit3", filename="{MODEL FILENAME}.zip", )
e6904d1
1
2
render_mode: rgb_array
