2021年10月6日
Pytorch-暑假实验的一些方法记录
摘要： 1.继承Dataset构建的类，按batch的最长句子数读取（而不是按训练集的最长，可以减少一丢丢显存） 以BiLSTM+CRF做分词所需的数据格式为例，重写collate_fn方法： 1 class SegDataset(Dataset): 2 def __init__(self, data_pa 阅读全文
posted @ 2021-10-06 20:51 最咸的鱼 阅读(37) 评论(0) 推荐(0) 编辑