TypeError: torch.nn.functional.relu is not a Module subclass

vision
1

I am trying to create a pytorch model and add relu activation after every linear layer. But I am getting

 TypeError : torch.nn.functional.relu is not a Module subclass

Model Class :

class Model(nn.Module):
    def __init__(self, in_features=4, out_features=3):
        super().__init__()
        
        self.net = nn.Sequential(
            nn.Linear(in_features, 8),
            F.relu,

            nn.Linear(8,9),
            F.relu,

            nn.Linear(9, out_features) )

    def forward(self,x):
       x = self.net(x)
       return x
2

nn.Sequential can only be populated with pytorch modules (nn.Module) class, and F.relu is just a function, not a module. You should change the F.relu in your code to nn.ReLU(), dont forget the parethesis :wink:

1 Like
3

thanks ! you saved my day