Skip to content

Conversation

@SeanPeppers
Copy link

This patch improves the fashion_mnist_demo.py pipeline for local federated learning by:

  • Adding last_accuracy to MyModule, satisfying the FLoX training API expectations
  • Setting TORCH_DATASETS=./data fallback to avoid subprocess crashes during test evaluation
  • Forcing spawn multiprocessing mode to avoid PyTorch fork() serialization issues

These changes make the example run cleanly and reproducibly on most systems.

Tested locally with 3 workers and 5 rounds using FedAvg.

…set fallback, updated dependencies and deprecated code
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant