def __init__()

in agents/bc_gan2.py [0:0]

• 20 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self,
                 state_dim,
                 action_dim,
                 max_action,
                 device,
                 z_dim=64,
                 w_dim=256,
                 num_layers=2):
        super(Generator, self).__init__()

        self.z_dim = z_dim
        self.device = device
        self.max_action = max_action

        self.network = nn.Sequential(nn.Linear(state_dim + action_dim, 256),
                                     nn.LeakyReLU(0.1),
                                     nn.Linear(256, 256),
                                     nn.LeakyReLU(0.1),
                                     nn.Linear(256, 256),
                                     nn.LeakyReLU(0.1),
                                     nn.Linear(256, action_dim),
                                     nn.Tanh())