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)