rmqn.py 文件源码

def __init__(self, input_num, action_num, max_buff_size, m, e):
        print("RMQN Model", input_num, action_num)
        super(RMQN, self).__init__(
            memory_module = MemoryModule(max_buff_size=max_buff_size, m=m, e=e),
            encoder=L.Linear(in_size=input_num, out_size=e),
            context=L.LSTM(in_size=e, out_size=m),
            quality=QualityPhi(m, action_num),
        )