Function _parse_and_to_tensor

generate_samples.py:68–73

(text, img_size=256, query_template='{}')

    return model

def _parse_and_to_tensor(text, img_size=256, query_template='{}'):
    tokenizer = get_tokenizer()
    text = query_template.format(*text.split('\t'))
    seq = tokenizer.parse_query(text, img_size=img_size)
    seq = torch.cuda.LongTensor(seq)
    return seq

def get_context(args, query_template='{}'):
    tokenizer = get_tokenizer()