search

datamllab / rlcard

Reinforcement Learning / AI Bots in Card (Poker) Games - Blackjack, Leduc, Texas, DouDizhu, Mahjong, UNO.
http://www.rlcard.org
MIT License
2.87k stars 619 forks source link

How to save agent? #304

Open Najeeb-00 opened 11 months ago

Najeeb-00 commented 11 months ago

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.

JocyeI commented 9 months ago

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