def gettemplate()

in models/decoders/voxel1.py [0:0]


def gettemplate(templatetype, **kwargs):
    if templatetype == "conv":
        return ConvTemplate(**kwargs)
    elif templatetype == "affinemix":
        return LinearTemplate(**kwargs)
    else:
        return None