e38d5d92c6ec69efbe03034fe659c88d2815e7cc,skorch/callbacks/training.py,Checkpoint,on_epoch_end,#Checkpoint#Any#,80

Before Change

        if do_checkpoint:
            target = self.target
            if isinstance(self.target, str):
                target = self.target.format(
                    net=net,
                    last_epoch=net.history[-1],
                    last_batch=net.history[-1, "batches", -1],
                )
            self._sink("Checkpoint! Saving model to {}.".format(target), net.verbose)
            net.save_params(target)

        net.history.record("event_cp", bool(do_checkpoint))

After Change

                    "validation scores for checkpointing.".format(e.args[0]))

        if do_checkpoint:
            self.save_model(net)
            self._sink("A checkpoint was triggered in epoch {}.".format(
                len(net.history) + 1
            ), net.verbose)

Instances