def __init__()

in experiments/grasp_stability/train.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

    def __init__(self, fields):
        super(Model, self).__init__()

        self.fields = fields

        for k in self.fields:
            # Load base network
            net = self.get_base_net()
            net_name = "net_{}".format(k)

            # Add for training
            self.add_module(net_name, net)

        self.nb_feature = 512
        self.fc1 = nn.Linear(self.nb_feature * len(fields), 512)
        self.fc2 = nn.Linear(512, 256)
        self.fc3 = nn.Linear(256, 2)