File size: 336 Bytes
803ef9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from .contrastive import Contrastive
from .w_mse import WMSE
from .byol import BYOL
METHOD_LIST = ["contrastive", "w_mse", "byol"]
def get_method(name):
assert name in METHOD_LIST
if name == "contrastive":
return Contrastive
elif name == "w_mse":
return WMSE
elif name == "byol":
return BYOL
|