Rong Ou
a343ae3b34
fix dupliate gpu check ( #9578 )
2023-09-14 05:53:46 +08:00
Rong Ou
0f35493b65
Add GPU support to NVFlare demo ( #9552 )
2023-09-06 17:03:59 +08:00
Sean Yang
12fe2fc06c
Fix federated learning demos and tests ( #9488 )
2023-08-16 15:25:05 +08:00
Jiaming Yuan
275da176ba
Document for device ordinal. ( #9398 )
...
- Rewrite GPU demos. notebook is converted to script to avoid committing additional png plots.
- Add GPU demos into the sphinx gallery.
- Add RMM demos into the sphinx gallery.
- Test for firing threads with different device ordinals.
2023-07-22 15:26:29 +08:00
Rong Ou
250b22dd22
Fix nvflare horizontal demo ( #9124 )
2023-05-05 16:48:22 +08:00
Rong Ou
fb941262b4
Add demo for vertical federated learning ( #9103 )
2023-04-28 16:03:21 +08:00
Jiaming Yuan
6a892ce281
Specify src path for isort. ( #8867 )
2023-03-06 17:30:27 +08:00
Rong Ou
0caf2be684
Update NVFlare demo to work with the latest release ( #8576 )
2022-12-09 02:48:20 +08:00
Jiaming Yuan
0d3da9869c
Require isort on all Python files. ( #8420 )
2022-11-08 12:59:06 +08:00
Rong Ou
b3208aac4e
Fix NVFLARE demo ( #8340 )
2022-10-14 12:18:34 +08:00
Rong Ou
668b8a0ea4
[Breaking] Switch from rabit to the collective communicator ( #8257 )
...
* Switch from rabit to the collective communicator
* fix size_t specialization
* really fix size_t
* try again
* add include
* more include
* fix lint errors
* remove rabit includes
* fix pylint error
* return dict from communicator context
* fix communicator shutdown
* fix dask test
* reset communicator mocklist
* fix distributed tests
* do not save device communicator
* fix jvm gpu tests
* add python test for federated communicator
* Update gputreeshap submodule
Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
2022-10-05 14:39:01 -08:00
Rong Ou
6eb23353d7
Update nvflare demo for release 2.1.2 ( #8038 )
2022-06-29 17:58:06 +08:00
Rong Ou
31e6902e43
Support GPU training in the NVFlare demo ( #7965 )
2022-06-02 21:52:36 +08:00
Rong Ou
af907e2d0d
Demo of federated learning using NVFlare ( #7879 )
...
Co-authored-by: jiamingy <jm.yuan@outlook.com>
2022-05-14 22:45:41 +08:00
