bacow/easy-rl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
JohnJim0816
2021-03-13 11:51:51 +08:00
parent 4df3169142
commit e522ba5510
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
codes/common/model.py
View File

@@ -5,20 +5,23 @@ Author: John
Email: johnjim0816@gmail.com Email: johnjim0816@gmail.com
Date: 2021-03-12 21:14:12 Date: 2021-03-12 21:14:12
LastEditor: John LastEditor: John
LastEditTime: 2021-03-12 21:28:46 LastEditTime: 2021-03-13 11:51:38
Discription: Discription:
Environment: Environment:
''' '''
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
class MLP(nn.Module): class MLP(nn.Module):
''' 全连接网络''' ''' 多层感知机
def __init__(self,state_dim): 输入：state维度
输出：概率
'''
def __init__(self,n_states,hidden_dim = 36):
super(MLP, self).__init__() super(MLP, self).__init__()
# 24和36为hidden layer的层数，可根据state_dim, n_actions的情况来改变 # 24和36为hidden layer的层数，可根据state_dim, n_actions的情况来改变
self.fc1 = nn.Linear(state_dim, 36) self.fc1 = nn.Linear(n_states, hidden_dim)
self.fc2 = nn.Linear(36, 36) self.fc2 = nn.Linear(hidden_dim,hidden_dim)
self.fc3 = nn.Linear(36, 1) # Prob of Left self.fc3 = nn.Linear(hidden_dim, 1) # Prob of Left
def forward(self, x): def forward(self, x):
x = F.relu(self.fc1(x)) x = F.relu(self.fc1(x))