![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pg-net阅读理解模型
爱喝喜茶爱吃烤冷面的小黑黑
这个作者很懒,什么都没留下…
展开
-
PG-net模型训练部分
注意:在这里最后几个epoch才优化coverage_loss import os import args import torch import random from tqdm import tqdm import evaluate from optimizer import BertAdam from dataset.dataloader import Dureader from dataset.file_utils import PYTORCH_PRETRAINED_BERT_CACHE from原创 2021-09-20 15:29:24 · 517 阅读 · 0 评论 -
pg-net训练模型部分模型部分
模型结构图: 代码流程图: 代码与输入输出: import sys import os import args import torch sys.path.append('.') from model_dir.modeling import BertPreTrainedModel,BertModel import torch.nn as nn from torch.nn import CrossEntropyLoss from transformer.models import Decoder,Decod原创 2021-09-18 16:44:22 · 331 阅读 · 0 评论