如何摆脱 GPT-2 警告消息?

每次运行 GPT-2 时,我都会收到此消息。有什么办法可以让它消失吗?

Some weights of GPT2LMHeadModel were not initialized from the model checkpoint at gpt2 and are newly initialized: ['h.0.attn.masked_bias', 'h.1.attn.masked_bias', 'h.2.attn.masked_bias', 'h.3.attn.masked_bias', 'h.4.attn.masked_bias', 'h.5.attn.masked_bias', 'h.6.attn.masked_bias', 'h.7.attn.masked_bias', 'h.8.attn.masked_bias', 'h.9.attn.masked_bias', 'h.10.attn.masked_bias', 'h.11.attn.masked_bias', 'lm_head.weight']
You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.


函数式编程
浏览 140回答 1
1回答

Smart猫小萌

是的,您需要在从转换器库导入任何内容之前更改日志级别:import logginglogging.basicConfig(level='ERROR')from transformers import GPT2LMHeadModelmodel = GPT2LMHeadModel.from_pretrained('gpt2')
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python