bojone / SPACES

端到端的长本文摘要模型(法研杯2020司法摘要赛道)
388 stars 91 forks source link

实验环境与训练速度 #34

Open chong0412 opened 2 years ago

chong0412 commented 2 years ago

按照给的实验环境(tensorflow 1.14 + keras 2.3.1 + bert4keras 0.9.7)训练生成式摘要时,在实验室服务器上(3090)跑得比自己笔记本上的还要慢,不知道是为什么,想请教一下~

bojone commented 2 years ago

我也不知道为什么

tianzuishiwo commented 2 years ago

3090 A100 采取了新的架构,支持并不好,但是也有补丁方案。参考这两篇文章: https://blog.csdn.net/qq_39543404/article/details/112171851 https://blog.csdn.net/wu496963386/article/details/109583045

bojone commented 2 years ago

3090 A100 采取了新的架构,支持并不好,但是也有补丁方案。参考这两篇文章: https://blog.csdn.net/qq_39543404/article/details/112171851 https://blog.csdn.net/wu496963386/article/details/109583045

3090、A100最佳的tf版本就是1.15,nvidia自己推荐的。

https://developer.nvidia.com/zh-cn/blog/accelerating-tensorflow-on-a100-gpus/

tianzuishiwo commented 2 years ago

嗯,苏神的demo tf-gpu1.15可以跑通。苏神请教一下,抽取模型结果正常,生成模型结果是重复的词,知道怎么回事吗 image

tianzuishiwo commented 2 years ago

原文:[1805] 唐鲜明与何伟华、深圳市华名威电汽车服务有 抽取句子:[469] 唐鲜明与何伟华、深圳市华名威电汽车服务有限公司侵权责任纠纷一审民事判决书上列原告唐鲜明诉被告何伟华、深圳市华名威电汽车服务有限公司(以下简称华名威公司)侵权责任纠纷一案,1、判决被告赔偿原告163755.57元,医疗费27086.37元、住院伙食补助费2500元、营养费4000元、护理费4598元、误工费26851元、残疾赔偿金46677.20元、精神损害抚慰金20000元、后续治疗费10000元 生成摘要:[1024] 告诉原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告原告