def graceful_shutdown()

in use-cases/model-fine-tuning-pipeline/fine-tuning/pytorch/src/fine_tune.py [0:0]


def graceful_shutdown(signal_number, stack_frame):
    signal_name = signal.Signals(signal_number).name

    logger.info(f"Received {signal_name}({signal_number}), shutting down...")
    # TODO: Add logic to handled checkpointing if required
    sys.exit(0)