Can you please try this?
import torch
from torch import nn
from torch.nn.utils import spectral_norm


class ReLULayer(nn.Module):
    def __init__(self, n_in, n_out):
        super(ReLULayer, self).__init__()
        self.n_in = n_in
        self.n_out = n_out
        self.linear = spectral_norm(nn.Linear(n_in, n_out)) #spectral norm in Linear Layer
        self.relu = nn.ReLU()

    def forward(self, input):
        output = self.linear(input)
        output = self.relu(output)
        return output


Also, check this link documentation.