Fix quickstart-fastai example (#2111)

adap · Jul 13, 2023 · 344c242 · 344c242
1 parent 982e3fd
commit 344c242
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/quickstart_fastai/client.py b/examples/quickstart_fastai/client.py
@@ -14,7 +14,7 @@
 
 # Load dataset
 dls = ImageDataLoaders.from_folder(
-    path, valid_pct=0.5, train="training", valid="testing"
+    path, valid_pct=0.5, train="training", valid="testing", num_workers=0
 )
 
 # Define model
@@ -24,7 +24,7 @@
 # Define Flower client
 class FlowerClient(fl.client.NumPyClient):
     def get_parameters(self, config):
-        return [val.numpy() for _, val in learn.model.state_dict().items()]
+        return [val.cpu().numpy() for _, val in learn.model.state_dict().items()]
 
     def set_parameters(self, parameters):
         params_dict = zip(learn.model.state_dict().keys(), parameters)