Skip to content

Instantly share code, notes, and snippets.

@s-gv
s-gv / model.py
Last active May 25, 2021 12:54
pong-ppo
import numpy as np
import torch
from torch import nn
from torch.nn import functional as F
from PIL import Image
import random
class Policy(nn.Module):
def __init__(self):
super(Policy, self).__init__()