nvflare-convert-pytorch

Installation
SKILL.md

NVFLARE Convert PyTorch

Use When

Use when converting an existing plain PyTorch training script, torch.nn.Module, manual training loop, state_dict workflow, data loader, checkpoint, or metric loop into an NVFLARE federated training job. Supports horizontal FL, Client API model exchange with FLModel, recipe aggregator= hooks, validation, and export.

Do Not Use When

Installs
6
Repository
nvidia/skills
GitHub Stars
3.4K
First Seen
6 days ago
nvflare-convert-pytorch — nvidia/skills