Spaces:
Running
on
Zero
Running
on
Zero
File size: 639 Bytes
9e15541 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from .mlp import ImplicitNet
from .resnetfc import ResnetFC
def make_mlp(conf, d_in, d_latent=0, allow_empty=False, **kwargs):
mlp_type = conf.get("type", "mlp") # mlp | resnet
if mlp_type == "mlp":
net = ImplicitNet.from_conf(conf, d_in + d_latent, **kwargs)
elif mlp_type == "resnet":
net = ResnetFC.from_conf(conf, d_in, d_latent=d_latent, **kwargs)
elif mlp_type == "resnet2":
net = ResnetFC.from_conf2(conf, d_in, d_latent=d_latent, **kwargs)
elif mlp_type == "empty" and allow_empty:
net = None
else:
raise NotImplementedError("Unsupported MLP type")
return net
|