Commit b2a8b263 authored by benkyoujouzu's avatar benkyoujouzu
Browse files

Add missing support for linear activation in hypernetwork

parent 737eb28f

modules/hypernetworks/hypernetwork.py

+1 −0
Original line number Diff line number Diff line
@@ -25,6 +25,7 @@ from statistics import stdev, mean 
class HypernetworkModule(torch.nn.Module):

    multiplier = 1.0

    activation_dict = {

        "linear": torch.nn.Identity,

        "relu": torch.nn.ReLU,

        "leakyrelu": torch.nn.LeakyReLU,

        "elu": torch.nn.ELU,

CRA Git | Maintained and supported by SUSTech CRA and CCSE