def METHOD_NAME(self, trainer):
    if is_master():
        trainer.logger.info('Saving checkpoint at final compress')
    cur_save_name = os.path.join(self.save_dir, 'compress_model.pth')
    save_checkpoint(trainer.model, cur_save_name, trainer.optimizer)