def main()

in ssiog/training.py [0:0]


def main():
    logger.info("testing....logger")
    try:
        success = True
        training()
    except Exception as e:
        success = False
        logger.error(f"Workload failed with error: {e}")
        traceback.print_exc()
        sys.exit(1)
    finally:
        # Make sure the flush the metrics in the buffer.
        close_metrics_logger()
        td.destroy_process_group()
        if success:
            logger.info("Workload completed successfully.")
            sys.exit(0)