9e449772c5f97f66c507cfe08b4d30b2b3a69316,vislogger/pytorchexperimentlogger.py,PytorchExperimentLogger,load_best_checkpoint,#Any#,169

Before Change

        name = "checkpoint_best.pth.tar"
        checkpoint_file = os.path.join(dir, name)

        if os.path.exists(checkpoint_file):
            return PytorchPlotLogger.load_checkpoint(checkpoint_file, **kwargs)
        else:
            return PytorchPlotLogger.load_lastest_checkpoint(dir=dir)

After Change

            dir_=dir_, name="checkpoint_best.pth.tar", **kwargs)

        if best_checkpoint is not None:
            return best_checkpoint
        else:
            return PytorchPlotLogger.load_lastest_checkpoint(dir_=dir_, **kwargs)

Instances