/third_party/mindspore/tests/st/fl/cross_silo_lenet/ |
D | test_cross_silo_lenet.py | 67 client_batch_size = args.client_batch_size variable 90 "client_batch_size": client_batch_size, 114 start_fl_job = StartFLJob(dataset.get_dataset_size() * args.client_batch_size) 117 data = Tensor(np.random.rand(client_batch_size, 3, 32, 32).astype(np.float32)) 118 label = Tensor(np.random.randint(0, 61, (client_batch_size)).astype(np.int32))
|
D | run_cross_silo_lenet_worker.py | 43 client_batch_size = args.client_batch_size variable 71 cmd_worker += " --client_batch_size=" + str(client_batch_size)
|
D | run_cross_silo_lenet_server.py | 55 client_batch_size = args.client_batch_size variable 88 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
/third_party/mindspore/tests/st/fl/cross_silo_femnist/ |
D | run_cross_silo_femnist_worker.py | 44 client_batch_size = args.client_batch_size variable 72 cmd_worker += " --client_batch_size=" + str(client_batch_size)
|
D | test_cross_silo_femnist.py | 84 client_batch_size = args.client_batch_size variable 109 "client_batch_size": client_batch_size, 317 …dataset = create_dataset_from_folder(train_path, args.img_size, args.client_batch_size, args.repea… 329 start_fl_job = StartFLJob(dataset.get_dataset_size() * args.client_batch_size) 344 train_acc, _ = evalute_process(model, train_path, args.img_size, args.client_batch_size) 345 test_acc, _ = evalute_process(model, test_path, args.img_size, args.client_batch_size)
|
D | run_cross_silo_femnist_server.py | 56 client_batch_size = args.client_batch_size variable 90 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
D | run_cross_silo_fasterrcnn_server.py | 57 client_batch_size = args.client_batch_size variable 92 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
D | test_fl_fasterrcnn.py | 56 client_batch_size = config.client_batch_size variable 81 "client_batch_size": client_batch_size,
|
/third_party/mindspore/tests/st/fl/albert/ |
D | run_hybrid_train_server.py | 68 client_batch_size = args.client_batch_size variable 109 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
D | cloud_train.py | 98 client_batch_size = args.client_batch_size 130 "client_batch_size": client_batch_size,
|
/third_party/mindspore/tests/st/fl/hybrid_lenet/ |
D | run_hybrid_train_server.py | 68 client_batch_size = args.client_batch_size variable 111 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
D | run_server_disaster_recovery.py | 76 client_batch_size = args.client_batch_size variable 129 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
D | test_hybrid_train_lenet.py | 79 client_batch_size = args.client_batch_size variable 111 "client_batch_size": client_batch_size,
|
/third_party/mindspore/tests/st/fl/mobile/ |
D | run_mobile_server.py | 72 client_batch_size = args.client_batch_size variable 115 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
D | test_mobile_lenet.py | 79 client_batch_size = args.client_batch_size variable 110 "client_batch_size": client_batch_size,
|
D | run_server_disaster_recovery.py | 74 client_batch_size = args.client_batch_size variable 125 cmd_server += " --client_batch_size=" + str(client_batch_size)
|
/third_party/mindspore/mindspore/ccsrc/ps/ |
D | ps_context.cc | 392 void PSContext::set_client_batch_size(uint64_t client_batch_size) { client_batch_size_ = client_bat… in set_client_batch_size() argument 394 uint64_t PSContext::client_batch_size() const { return client_batch_size_; } in client_batch_size() function in mindspore::ps::PSContext
|
D | ps_context.h | 156 void set_client_batch_size(uint64_t client_batch_size); 157 uint64_t client_batch_size() const;
|
/third_party/mindspore/mindspore/parallel/ |
D | _ps_context.py | 103 "client_batch_size": ps_context().client_batch_size,
|
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/ |
D | start_fl_job_kernel.cc | 242 fl_plan_builder.add_mini_batch(SizeToInt(ps::PSContext::instance()->client_batch_size())); in BuildStartFLJobRsp()
|
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
D | init.cc | 404 ….def("client_batch_size", &PSContext::client_batch_size, "Get federated learning client batch size… in PYBIND11_MODULE()
|
/third_party/mindspore/mindspore/ccsrc/fl/server/ |
D | server.cc | 588 response["client_batch_size"] = ps::PSContext::instance()->client_batch_size(); in HandleQueryInstanceRequest()
|