Downloads · 30 days
0
bvk1ng/atari_breakout_v0_dqn
atari_breakout_v0_dqn is a reinforcement learning model from bvk1ng. Use it for the reinforcement learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
The agent showcased in this space is trained using the Deep Q-Learning algorithm. The agent was trained for $$3500$$ episodes with a learning rate of $$0.00001$$ and an epsilon value that decreased linearly over time.
Downloads · 30 days
0
Access
Public
Updated Dec 28, 2023
Repo size
6.7 MB
Likes
0
Public
Click a slice to open those files.
.pt6.7 MB · 99%
From the Hugging Face model README
The agent showcased in this space is trained using the Deep Q-Learning algorithm. The agent was trained for $$3500$$ episodes with a learning rate of $$0.00001$$ and an epsilon value that decreased linearly over time.
python main.py --model_folder <Name of the folder> --model_name <Name of the model> --save_video 1 --video_name <Name of the video file>