NVFLARE Convert PyTorch

Use When

Use when the user asks to convert an existing PyTorch training script, torch.nn.Module, state_dict workflow, data loader, checkpoint, or metric loop into an NVFLARE federated training job.

Do Not Use When

Do not use for PyTorch Lightning, Hugging Face Trainer, TensorFlow, XGBoost, scikit-learn, Kubernetes deployment, production submission, or generic PyTorch debugging that does not ask for FLARE conversion.

Workflow