Usage
Each benchmark is compressed as a .zip file. To use benchmarks here, please follow the three steps below
-
Download the benchmark .zip file
-
Decompress the .zip file into the currently working project directory
-
Use the decompressed directory as a python module, and generate federated task from it
Example on MNIST
-
download mnist_classification.zip from here and decompress it into the currently working project directory.
-
Write codes as follows to use it
import flgo
import mnist_classification
task = './test_mnist_download'
flgo.gen_task({'benchmark': mnist_classification}, task)
import flgo.algorithm.fedavg as fedavg
flgo.init(task, fedavg, {'gpu':0}).run()