DQN003.py 文件源码

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, NUM_ACTIONS)
        )