fluke.distr.server
¶
Classes included in fluke.distr.server
Server for parallel federated learning using multiple GPUs. |
class fluke.distr.server.ParallelServer
- class fluke.distr.server.ParallelServer(*args: Any, **kwargs: dict[str, Any])[source]¶
Server for parallel federated learning using multiple GPUs.
This class is a subclass of
Server
that allows the use of multiple GPUs - one for each client. It uses multiprocessing to handle the training of clients in parallel. It is designed to work with the torch.multiprocessing module, which allows for parallel execution of client training processes across multiple GPUs.