def __init__(self, args): super().__init__() self.args = args self.encoder = Encoder(args=args, bi_awd_layer="first", architecture="both") self.decoder = Decoder(args=args, in_size=args.n_hid * 2 + 300)
def __init__(self, args): super().__init__() self.args = args self.encoder = Encoder(args, bi_awd_layer="second", architecture="before") self.decoder = Decoder(args, in_size=args.n_hid * 2)