Additional parameters in def__init__ for Optuna Trial

Question

class ConvolutionalNetwork(nn.Module):
    def __init__(self, in_features):
        super().__init__()
        self.in_features = in_features
        # this computes