error for load_partition_data_mnist()

How to spedify batch_size in terminal?

Author: GuokaiLiu

python/examples/simulation/sp_fedavg_mnist_lr_example/torch_fedavg_mnist_lr_custum_data_and_model_example.py

@@ -23,7 +23,7 @@ def load_data(args):
  test_data_local_dict,
  class_num,
  ) = load_partition_data_mnist(
- args.batch_size,
+ args, 64, # [Origin]: args.batch_size, [Now]: args, 64 [Note]: batch_size=64
  train_path=args.data_cache_dir + "MNIST/train",
  test_path=args.data_cache_dir + "MNIST/test",
  )