首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >当我构建一个自动生成源代码注释的深度学习模型时。我得到了下面的错误

当我构建一个自动生成源代码注释的深度学习模型时。我得到了下面的错误
EN

Stack Overflow用户
提问于 2021-06-03 03:17:10
回答 1查看 37关注 0票数 0

当我构建一个自动生成源代码注释的深度学习模型时。我得到了上面的错误。我使用了转换器模型来创建model.Can,谁能给我一个解决方案

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-06-03 03:41:09

您似乎正在使用transformers模块。根据文档,提供参数的顺序为:

代码语言:javascript
复制
 def __init__(
        self,
        vocab_size=267735,
        cutoffs=[20000, 40000, 200000],
        d_model=1024,
        d_embed=1024,
        n_head=16,
        d_head=64,
        d_inner=4096,
        div_val=4,
        pre_lnorm=False,
        n_layer=18,
        mem_len=1600,
        clamp_len=1000,
        same_length=True,
        proj_share_all_but_first=True,
        attn_type=0,
        sample_softmax=-1,
        adaptive=True,
        dropout=0.1,
        dropatt=0.0,
        untie_r=True,
        init="normal",
        init_range=0.01,
        proj_init_std=0.01,
        init_std=0.02,
        layer_norm_epsilon=1e-5,
        eos_token_id=0,
        **kwargs
    )

因为您没有为构造函数提供正确的名称,所以它将n_x_vocab关联到vocab_size,将n_y_vocab关联到cutoffs。因此,所需的参数n_headd_innern_layerd_head被认为是缺少的。

一旦您以适当的顺序提供了参数和/或显式地命名了参数,就像您在d_model = 256中所做的那样,这个错误应该被修复。

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67811224

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档