frmqn.py 文件源码

def __init__(self, input_num, action_num, max_buff_size, m, e):
        assert(m == e)
        print("FRMQN Model", input_num, action_num)
        super(FRMQN, 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+m), out_size=m),
            quality=QualityPhi(m, action_num),
        )
        self.o = None