Generator Error

[t1804330987]

class Generator(nn.Module):
"""Fake Label Generator in SemiReward"""

def __init__(self, feature_dim=384):
    super(Generator, self).__init__()
    self.fc_layers = nn.Sequential(
        nn.Linear(feature_dim, 256),
        nn.ReLU(),
        nn.Linear(256, 128),
        nn.ReLU(),
        nn.Linear(128, 64),
        nn.ReLU(),
        nn.Linear(64, 1)  # Error  , Alter : nn.Linear(64, num_classes) 
    )

def forward(self, x):
    x = self.fc_layers(x)
    x = F.relu(x)
    return x