Reinforcement Learning, Gaming, Deep Learning.
GradientPolicy(nn.Module): def __init__(self, in_features, out_dims, hidden_size=128): super().__init__() self.fc1 = nn.Linear(in_features, hidden_size) self.fc2 = nn.Linear(hidden_size, hidden_size) self.fc_mu = nn.Linear(hidden_size, out_dims) self.fc_std = nn.Linear(hidden_size, out_dims) def