def __init__(self, config):
super().__init__()
self.decoder = Decoder.build({**config["decoder"], "rename": True})
self.soft_mem_mask = config["decoder"]["mem_mask"] == "soft"
if self.soft_mem_mask:
self.mem_encoder = Encoder.build(config["mem_encoder"])
self.mem_decoder = Decoder.build(config["mem_decoder"])
self.decoder.mem_encoder = self.mem_encoder
self.decoder.mem_decoder = self.mem_decoder
self.beam_size = config["test"]["beam_size"]
def __init__(self, config):
super().__init__()
self.decoder = Decoder.build({**config["decoder"]})
self.subtype = config["decoder"]["type"] in ["XfmrSubtypeDecoder"]
self.soft_mem_mask = config["decoder"]["mem_mask"] == "soft"
if self.soft_mem_mask:
self.mem_encoder = Encoder.build(config["mem_encoder"])
self.mem_decoder = Decoder.build(config["mem_decoder"])
self.decoder.mem_encoder = self.mem_encoder
self.decoder.mem_decoder = self.mem_decoder
self.beam_size = config["test"]["beam_size"]
