修改模型词嵌入的输入，（vocab_size, embedding_size）m使得模型可以解决多个任务，主要是vocab_size这个

对于交叉熵损失，模型的输出类型需要为torch.float32,而标签的类型也需要为torch.float32