Pranay Dilip Salkar comments

Repositories
Issues
Comments

Results 1 comments of


                                            Pranay Dilip Salkar

TypeError: linear(): argument 'input' (position 1) must be Tensor, not ReLU

import torch import torch.nn as nn class YourModel(nn.Module): def __init__(self, input_size, hidden_size, output_size): super(YourModel, self).__init__() self.l1 = nn.Linear(input_size, hidden_size) self.relu = nn.ReLU() # ReLU activation function self.l2 = nn.Linear(hidden_size, output_size)...