Skip to content

Instantly share code, notes, and snippets.

@elumixor

elumixor/TRPO_actor.py

Created May 24, 2020 22:28
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save elumixor/b25736cf4f6e69d26d4aee7c71adb8aa to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save elumixor/b25736cf4f6e69d26d4aee7c71adb8aa to your computer and use it in GitHub Desktop.
Download ZIP
Medium TRPO Actor
Raw
TRPO_actor.py
import torch.nn as nn
actor_hidden = 32
actor = nn.Sequential(nn.Linear(state_size, actor_hidden),
nn.ReLU(),
nn.Linear(actor_hidden, num_actions),
nn.Softmax(dim=1))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment