DQN PYTORCH 玩FlappyBird

最新推荐文章于 2024-03-09 21:59:12 发布

原创

最新推荐文章于 2024-03-09 21:59:12 发布 · 462 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #pygame

import torch
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
# import gym

from ple.games.flappybird import FlappyBird
from ple import PLE
from pygame.constants import K_w
import time
import random
import collections
import numpy as np
import os

# Hyper Parameters
BATCH_SIZE = 32
LR = 0.0001                 # learning rate
EPSILON = 0.9               # greedy policy
GAMMA = 0.999                 # reward discount
TARGET_REPLACE_ITER = 100   # target update frequency
MEMORY_CAPACITY = 20000

game = FlappyBird()
env = PLE(game, fps=30, display_screen=True)

N_ACTIONS =2# env.action_space.n
N_STATES = 8#env.observation_space.shape[0]


class Net(nn.Module):
    def __init__(self, ):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(N_STATES, 128)
        self.fc1.weight.data.normal_(0, 0.