def __init__(self, state_dim = STATE_DIM):
super(Q, self).__init__(
l1=F.Linear(state_dim, 50),
l2=F.Linear(50, 50),
q_value=F.Linear(50, 5)
)
微信
公众号
扫码关注公众号