Home
last modified time | relevance | path

Searched refs:client_batch_size (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/tests/st/fl/cross_silo_lenet/
Dtest_cross_silo_lenet.py67 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))
Drun_cross_silo_lenet_worker.py43 client_batch_size = args.client_batch_size variable
71 cmd_worker += " --client_batch_size=" + str(client_batch_size)
Drun_cross_silo_lenet_server.py55 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/
Drun_cross_silo_femnist_worker.py44 client_batch_size = args.client_batch_size variable
72 cmd_worker += " --client_batch_size=" + str(client_batch_size)
Dtest_cross_silo_femnist.py84 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)
Drun_cross_silo_femnist_server.py56 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/
Drun_cross_silo_fasterrcnn_server.py57 client_batch_size = args.client_batch_size variable
92 cmd_server += " --client_batch_size=" + str(client_batch_size)
Dtest_fl_fasterrcnn.py56 client_batch_size = config.client_batch_size variable
81 "client_batch_size": client_batch_size,
/third_party/mindspore/tests/st/fl/albert/
Drun_hybrid_train_server.py68 client_batch_size = args.client_batch_size variable
109 cmd_server += " --client_batch_size=" + str(client_batch_size)
Dcloud_train.py98 client_batch_size = args.client_batch_size
130 "client_batch_size": client_batch_size,
/third_party/mindspore/tests/st/fl/hybrid_lenet/
Drun_hybrid_train_server.py68 client_batch_size = args.client_batch_size variable
111 cmd_server += " --client_batch_size=" + str(client_batch_size)
Drun_server_disaster_recovery.py76 client_batch_size = args.client_batch_size variable
129 cmd_server += " --client_batch_size=" + str(client_batch_size)
Dtest_hybrid_train_lenet.py79 client_batch_size = args.client_batch_size variable
111 "client_batch_size": client_batch_size,
/third_party/mindspore/tests/st/fl/mobile/
Drun_mobile_server.py72 client_batch_size = args.client_batch_size variable
115 cmd_server += " --client_batch_size=" + str(client_batch_size)
Dtest_mobile_lenet.py79 client_batch_size = args.client_batch_size variable
110 "client_batch_size": client_batch_size,
Drun_server_disaster_recovery.py74 client_batch_size = args.client_batch_size variable
125 cmd_server += " --client_batch_size=" + str(client_batch_size)
/third_party/mindspore/mindspore/ccsrc/ps/
Dps_context.cc392 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
Dps_context.h156 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.py103 "client_batch_size": ps_context().client_batch_size,
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/round/
Dstart_fl_job_kernel.cc242 fl_plan_builder.add_mini_batch(SizeToInt(ps::PSContext::instance()->client_batch_size())); in BuildStartFLJobRsp()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc404 ….def("client_batch_size", &PSContext::client_batch_size, "Get federated learning client batch size… in PYBIND11_MODULE()
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dserver.cc588 response["client_batch_size"] = ps::PSContext::instance()->client_batch_size(); in HandleQueryInstanceRequest()