drqn.py 文件源码

def __init__(self, input_num, action_num):
        print("DRQN Model", input_num, action_num)
        super(DRQN, self).__init__(
            fc1=L.Linear(input_num, 256),
            lstm=L.LSTM(256, 256),
            fc2=L.Linear(256, action_num),
        )