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=args, awd_layer="second", architecture="before") self.decoder = Decoder(args=args, in_size=args.n_hid * 2)
def __init__(self, args): super().__init__() self.args = args self.encoder = Encoder(args) self.decoder = Decoder(args, in_size=2560)
def __init__(self, args): super().__init__() self.args = args self.encoder = Encoder(args, awd_layer="last") self.decoder = Decoder(args, in_size=args.n_hid * 2)
def __init__(self, args): super().__init__() self.args = args self.encoder = Encoder(args, awd_layer="second", project_size=300) self.decoder = Decoder(args, in_size=args.n_hid * 2)