模型训练模式,推理结果会一直变化
收藏
加载训练好的模型
model = BertModel.from_pretrained(task_path, pool_act='linear')
想继续训练的时候发现,当使用model.train()时,
model(input_ids=input_ids, token_type_ids=token_type_ids)
每次运行完全相同的数据都不一致,有人知道这是什么原因吗
0
收藏
请登录后评论