--- a +++ b/src/utils/cl_strategies.py @@ -0,0 +1,38 @@ +""" +Continual learning strategies. + +Contains: + +- Baselines +- Regularization methods +- Rehearsal methods +- Misc +""" + +from avalanche.training import Naive, JointTraining, Cumulative # Baselines +from avalanche.training import EWC, LwF, SynapticIntelligence # Regularisation +from avalanche.training import Replay, GDumb, GEM, AGEM # Rehearsal +from avalanche.training import AR1, CWRStar, CoPE, StreamingLDA + +STRATEGIES = { + # Baselines + "Naive": Naive, + "Naive_no_reg": Naive, + "Joint": JointTraining, + "Cumulative": Cumulative, + # Regularization based + "EWC": EWC, + "OnlineEWC": EWC, + "LwF": LwF, + "SI": SynapticIntelligence, #'LFL':LFL, + # Replay + "Replay": Replay, + "GEM": GEM, + "AGEM": AGEM, + "GDumb": GDumb, + "CoPE": CoPE, + # Misc. + "AR1": AR1, + "StreamingLDA": StreamingLDA, + "CWRStar": CWRStar, +}