domains / families / science / nvflare-convert-pytorch

nvflare-convert-pytorch

separate

Convert existing plain or manual PyTorch training code into an NVFLARE federated job using Client API model exchange, local validation, and job export; use when the user names plain PyTorch or preliminary source inspection identifies one plain-PyTorch owner, and not for Lightning, other frameworks, deployment, POC/production lifecycle, or experiment workflows.

When to use

• converting plain PyTorch training scripts
• converting torch.nn.Module workflows
• converting manual training loops
• converting state_dict workflows
• converting data loaders and checkpoints
• converting metric loops
• horizontal FL conversion
• Client API model exchange with FLModel

When not to use

• Do not use for PyTorch Lightning (route to `nvflare-convert-lightning`), Hugging Face Trainer (route to `nvflare-convert-huggingface`), TensorFlow, XGBoost, scikit-learn, failed jobs (route to `nvflare-diagnose-job`), federated statistics…

Skill page

Source