main_continual.py
is the main file. Run it each time with the subsequent hyperparameters configurations:
pretrained = False
will give youweights_1.pkl
(everything else as it is by default)pretrained = True
, andtask = 2
will give youweights_2.pkl
to evaluate forward transferis_training = False
,pretrained = True
,task = 2
to evaluate backward transfer, i.e. ability to overcome catastrophic forgettingpretrained = True
,task = 3
will give youweights_3.pkl
is_training = False
,pretrained = True
,task = 3
- (and so on for as many tasks as you want)