datamllab / rlcard

Reinforcement Learning / AI Bots in Card (Poker) Games - Blackjack, Leduc, Texas, DouDizhu, Mahjong, UNO.

Home Page:http://www.rlcard.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to save agent?

Najeeb-00 opened this issue · comments

I've made a simple tic-tac-toe game and trained a dqnAgent vs a random agent and got good results. But I don't know how to save the agent/ model to use it later.
I know this question must sound trivial but I could not find an answer on my own, I am a beginner in ML stuff and this was my first project. help is appreciated.

when training......torch.save(model_name)