Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- self.network = keras.models.Sequential()
- self.network.add(Conv2D(16, (3, 3), strides=2, input_shape=state_shape))
- self.network.add(LeakyReLU())
- self.network.add(Conv2D(32, (3, 3), strides=2))
- self.network.add(LeakyReLU())
- self.network.add(Conv2D(64, (3, 3), strides=2))
- self.network.add(LeakyReLU())
- self.network.add(Flatten())
- self.network.add(Dense(256))
- self.network.add(LeakyReLU())
- self.network.add(Dense(n_actions, activation='linear'))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement