def __init__(self, args): super().__init__() self.args = args self.encoder = Encoder(args, bi_awd_layer="second", project_size=300) self.decoder = Decoder(args, in_size=args.n_hid * 2)
def __init__(self, args): super().__init__() self.args = args self.encoder = Encoder(args, bi_awd_layer="last") self.decoder = Decoder(args, in_size=args.n_hid * 2)