Spaces:
Sleeping
Sleeping
File size: 2,051 Bytes
cb80c28 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
def get_model(model_name, args):
name = model_name.lower()
if name == "icarl":
from models.icarl import iCaRL
return iCaRL(args)
elif name == "bic":
from models.bic import BiC
return BiC(args)
elif name == "podnet":
from models.podnet import PODNet
return PODNet(args)
elif name == "lwf":
from models.lwf import LwF
return LwF(args)
elif name == "ewc":
from models.ewc import EWC
return EWC(args)
elif name == "wa":
from models.wa import WA
return WA(args)
elif name == "der":
from models.der import DER
return DER(args)
elif name == "finetune":
from models.finetune import Finetune
return Finetune(args)
elif name == "replay":
from models.replay import Replay
return Replay(args)
elif name == "gem":
from models.gem import GEM
return GEM(args)
elif name == "coil":
from models.coil import COIL
return COIL(args)
elif name == "foster":
from models.foster import FOSTER
return FOSTER(args)
elif name == "rmm-icarl":
from models.rmm import RMM_FOSTER, RMM_iCaRL
return RMM_iCaRL(args)
elif name == "rmm-foster":
from models.rmm import RMM_FOSTER, RMM_iCaRL
return RMM_FOSTER(args)
elif name == "fetril":
from models.fetril import FeTrIL
return FeTrIL(args)
elif name == "pass":
from models.pa2s import PASS
return PASS(args)
elif name == "il2a":
from models.il2a import IL2A
return IL2A(args)
elif name == "ssre":
from models.ssre import SSRE
return SSRE(args)
elif name == "memo":
from models.memo import MEMO
return MEMO(args)
elif name == "beefiso":
from models.beef_iso import BEEFISO
return BEEFISO(args)
elif name == "simplecil":
from models.simplecil import SimpleCIL
return SimpleCIL(args)
else:
assert 0
|