2022年4月5日
摘要: torch.nn.Embedding(num_embeddings, embedding_dim, padding_idx=None, max_norm=None, norm_type=2.0, scale_grad_by_freq=False, sparse=False, _weight=None 阅读全文
posted @ 2022-04-05 22:31 啥123 阅读(781) 评论(0) 推荐(0) 编辑
摘要: 1 编码方式 如果训练数据采用的编码方式是Bert,则通过BertTokenizer.from_pretrained(args.bert_model_dir)来加载预训练模型和训练的参数。 2 加载数据集 加载train_dataset训练数据,test_dataset测试数据,word_vocab 阅读全文
posted @ 2022-04-05 09:24 啥123 阅读(314) 评论(0) 推荐(0) 编辑