darkvgg.py

class DQN(object):
    def __init__(self, input, num_actions):
        self.conv1 = nn.Conv2d(input, 32, 3, stride=2, padding=1)
        self.conv2 = nn.Conv2d(input, 32, 3, stride=2, padding=1)
        self.conv3 = nn.Conv2d(input, 32, 3, stride=2, padding=1)
        self.linear = nn.Linear(256, num_actions)
        """ initialize weights"""
    def forward(self, inputs):       
        x = nn.elu(self.conv1(inputs))
        x = nn.elu(self.conv2(x))
        x = nn.elu(self.conv3(x))    
        x = x.view(-1, 32 * 3 * 3)
        return self.linear(x)