Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def __init__(self, input_size, nb_action):
- super(Network, self).__init__()
- self.input_size = input_size
- self.nb_action = nb_action
- self.fc1 = nn.Linear(input_size, 100)
- self.fc2 = nn.Linear(100, nb_action)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement