elumixor · May 24, 2020 22:32
diff --git a/TRPO_get_action.py b/TRPO_get_action.py
 from torch.distributions import Categorical


 def get_action(state):
    state = torch.tensor(state).float().unsqueeze(0)  # Turn state into a batch with a single element
    dist = Categorical(actor(state))  # Create a distribution from probabilities for actions
    return dist.sample().item()
	from torch.distributions import Categorical


	def get_action(state):
	state = torch.tensor(state).float().unsqueeze(0) # Turn state into a batch with a single element
	dist = Categorical(actor(state)) # Create a distribution from probabilities for actions
	return dist.sample().item()
No results found