Method build_checkpointer

projects/CenterNet/centernet/trainer.py:18–26

(cls, model, output_dir, optimizer, scheduler)

    @classmethod
    def build_checkpointer(cls, model, output_dir, optimizer, scheduler):
        checkpointer = CenterNetCheckpointer(
            # Assume you want to save checkpoints together with logs/statistics
            model,
            output_dir,
            optimizer=optimizer,
            scheduler=scheduler,
        )
        return checkpointer

    @classmethod
    def build_train_loader(cls, cfg):