def __init__(self, config, img_size, vis):
    super(Transformer, self).__init__()
    self.embeddings = Embeddings(config, img_size=img_size)
    self.encoder = Encoder(config, vis)
def __init__(self, config, img_size, vis):
    super(Transformer, self).__init__()
    self.embeddings = Embeddings(config, img_size=img_size)
    self.encoder = Encoder(config, vis)