Hyperparameter tuning with ray tune

vision
1

Hello! I’m trying to implement an EfficientNet-B7 in this code here Hyperparameter tuning with Ray Tune — PyTorch Tutorials 1.12.1+cu102 documentation
and it is not working because I don’t know how to put this ‘L1’ and ‘L2’:

class Net(nn.Module):
    def __init__(self, l1=120, l2=84):
        super(Net, self).__init__()
        self.conv1 = nn.Conv2d(3, 6, 5)
        self.pool = nn.MaxPool2d(2, 2)
        self.conv2 = nn.Conv2d(6, 16, 5)
        self.fc1 = nn.Linear(16 * 5 * 5, l1)
        self.fc2 = nn.Linear(l1, l2)
        self.fc3 = nn.Linear(l2, 10)

in the EfficientNet-B7. Also, how can I modify this part later using L1 and L2?

net = Net(config["l1"], config["l2"])