嘉渊
2023-04-25 ee9e86d5f0b5f546f77de1ea8b59cdb67aaad478
funasr/bin/train.py
@@ -527,6 +527,7 @@
                                                                   distributed_option.dist_rank,
                                                                   distributed_option.local_rank))
    logging.info(pytorch_cudnn_version())
    logging.info("Args: {}".format(args))
    logging.info(model_summary(model))
    logging.info("Optimizer: {}".format(optimizers))
    logging.info("Scheduler: {}".format(schedulers))