Skip to content
Snippets Groups Projects
Commit dac3eee5 authored by Egli Adrian (IT-SCI-API-PFI)'s avatar Egli Adrian (IT-SCI-API-PFI)
Browse files

Tensorboard support added

parent 6a2fd382
No related branches found
No related tags found
No related merge requests found
......@@ -34,7 +34,7 @@ class EpisodeBuffers:
class ActorCriticModel(nn.Module):
def __init__(self, state_size, action_size, device, hidsize1=128, hidsize2=128):
def __init__(self, state_size, action_size, device, hidsize1=512, hidsize2=256):
super(ActorCriticModel, self).__init__()
self.device = device
self.actor = nn.Sequential(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment