update

嘉渊

2023-04-25 95d6db2656e9e5c135e8d6af4027bf76a83ec998

update

 funasr/datasets/small_datasets/sequence_iter_factory.py

@@ -83,7 +83,7 @@
            args.max_update = len(bs_list) * args.max_epoch
            logging.info("Max update: {}".format(args.max_update))

        if args.distributed:
        if args.distributed and mode=="train":
            world_size = torch.distributed.get_world_size()
            rank = torch.distributed.get_rank()
            for batch in batches: