22d481f6a6ac88bea4ec64f97e99132e47336f63,qucumber/callbacks.py,ModelSaver,on_epoch_end,#ModelSaver#Any#Any#,71

Before Change

                metadata = self.metadata(rbm, epoch)
            elif isinstance(self.metadata, dict):
                metadata = self.metadata
            elif self.metadata is None:
                metadata = {}

            if self.metadata_only:
                torch.save(metadata, save_path)
            else:

After Change

    def on_epoch_end(self, rbm, epoch):
        if epoch % self.period == 0:
            save_path = os.path.join(self.path, self.file_name.format(epoch))
            self._save(rbm, epoch, save_path)


class Logger(Callback):
    Callback which logs output at regular intervals.

Instances