CHECKPOINTS#

Functions#

utils.checkpoints.mammoth_load_checkpoint(args, model, ignore_classifier=False)[source]#

Loads the keys from the given checkpoint. - Handles DataParallel and DistributedDataParallel checkpoints. - Handles checkpoints from previous versions of the code. - Handles head initialization for LUCIR.

Parameters:

  • args – the model with the checkpoint loaded.

  • model (Module) – the model to be loaded.

  • ignore_classifier – whether to ignore the classifier weights.

Returns:

the model with the checkpoint loaded.

Return type:

Module