TypeError: __init__() got an unexpected keyword argument 'batch_size'

在运行代码的过程中遇到以下报错：

Traceback (most recent call last):
  File "D:/Downloads/prnet-master/main.py", line 178, in 
    main()
  File "D:/Downloads/prnet-master/main.py", line 146, in main
    train_loader = DataLoader(ModelNet40(num_points=args.n_points,
TypeError: __init__() got an unexpected keyword argument 'batch_size'

报错部分代码：

   
def main():
    parser.add_argument('--batch_size', type=int, default=36, metavar='batch_size',
                        help='Size of batch)')


 if args.dataset == 'modelnet40':
        train_loader = DataLoader(ModelNet40(num_points=args.n_points,
                                             num_subsampled_points=args.n_subsampled_points,
                                             partition='train', gaussian_noise=args.gaussian_noise,
                                             unseen=args.unseen, rot_factor=args.rot_factor),
                                  batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=6)
        test_loader = DataLoader(ModelNet40(num_points=args.n_points,
                                            num_subsampled_points=args.n_subsampled_points,
                                            partition='test', gaussian_noise=args.gaussian_noise,
                                            unseen=args.unseen, rot_factor=args.rot_factor),
                                 batch_size=args.test_batch_size, shuffle=False, drop_last=False, num_workers=6)
    else:
        raise Exception("not implemented")

并且args中有batch-size

img

尝试了很多方法都没有解决，请问各位有什么解决办法吗？