Skip to content

Instantly share code, notes, and snippets.

@Paulescu

Paulescu/load_best_agent.py

Created March 4, 2022 10:21
Show Gist options
  • Save Paulescu/eb19cb583ac9f02ef1cc3df97f05686b to your computer and use it in GitHub Desktop.
Save Paulescu/eb19cb583ac9f02ef1cc3df97f05686b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_best_agent.py
from src.q_agent import QAgent
from src.config import SAVED_AGENTS_DIR
# you can find the agent_id for the best run in the MLflow
# dashboard.
# 298 is the value in my case, but you need to check what is your
agent_id = 298
path_to_saved_model = SAVED_AGENTS_DIR / 'CartPole-v1' / str(agent_id)
agent = QAgent.load_from_disk(env, path_to_saved_model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment