update codes

2021-12-28 18:46:52 +08:00
parent 41fb561d25
commit bd51b5a7ad
52 changed files with 305 additions and 292 deletions
--- a/codes/PolicyGradient/model.py
+++ b/codes/PolicyGradient/model.py
@@ -19,7 +19,7 @@ class MLP(nn.Module):
    '''
    def __init__(self,input_dim,hidden_dim = 36):
        super(MLP, self).__init__()
-        # 24和36为hidden layer的层数，可根据input_dim, n_actions的情况来改变
+        # 24和36为hidden layer的层数，可根据input_dim, action_dim的情况来改变
        self.fc1 = nn.Linear(input_dim, hidden_dim)
        self.fc2 = nn.Linear(hidden_dim,hidden_dim)
        self.fc3 = nn.Linear(hidden_dim, 1)  # Prob of Left