def get_callbacks()

in torchrecipes/audio/source_separation/train_app.py [0:0]

• 22 lines of code
• 1 McCabe index (conditional complexity)

    def get_callbacks(self) -> List[Callback]:
        """
        Override this method to return a list of callbacks to be passed
        into Trainer. You can add additional ModelCheckpoint here
        """
        checkpoint_dir = os.path.join(self.root_dir, "checkpoints")
        checkpoint = ModelCheckpoint(
            checkpoint_dir,
            monitor="Losses/val_loss",
            mode="min",
            save_top_k=5,
            save_weights_only=True,
            verbose=True,
        )
        callbacks = [
            checkpoint,
            EarlyStopping(
                monitor="Losses/val_loss", mode="min", patience=30, verbose=True
            ),
        ]
        callbacks = [
            EarlyStopping(
                monitor="Losses/val_loss", mode="min", patience=30, verbose=True
            ),
        ]
        return callbacks