当前位置: 代码网 > it编程>游戏开发>ar > NotImplementedError: Module is missing the required “forward“ function

NotImplementedError: Module is missing the required “forward“ function

2024年08月04日 ar 我要评论
2.def forward函数与def __init__(self,config):一定要对齐。1.重写父类函数时,函数名称写错,我将。

在做中文文本情感分析model类定义的时候报错如下:

有两种可能:

1.重写父类函数时,函数名称写错,我将forward写成了forword 最终导致程序报错:

import torch
import torch.nn as nn
import torch.nn.functional as f
import numpy as np

class model(nn.module):
    def __init__(self,config):
        super(model, self).__init__()
        self.embeding = nn.embedding(config.n_vocab, config.embed_size,padding_idx=config.n_vocab - 1)
        self.lstm = nn.lstm(config.embed_size, config.hidden_size,
config.num_layers,
bidirectional=true,batch_first=true,
dropout=config.dropout)
        self.maxpool = nn.maxpool1d(config.pad_size)
        self.fc = nn.linear(config.hidden_size * 2 + config.embed_size, config.num_classes)
        self.softmax = nn.softmax(dim=1)

    def forword(self,x):
        embed = self.embeding(x)
        out,_ = self.lstm(embed)
        out = torch.cat((embed,out),2)
        out = f.relu(out)
        out = out.permute(0,2,1)
        out = self.maxpool(out).reshape(out.size()[0],-1)
        out = self.fc(out)
        out = self.softmax(out)
        return out

if __name__ == "__main__":
    from configs import config

    cfg = config()
    cfg.pad_size = 640
    model_textcls = model(config = cfg)
    input_tensor = torch.tensor([i for i in range(640)]).reshape([1,640])
    out_tensor = model_textcls.forward(input_tensor)
    print(out_tensor.size())
    print(out_tensor)

2.def forward函数与def __init__(self,config):一定要对齐。

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com