metadata
tags:
- federated-learning
- shufflenet
- GTSRB
- krum
datasets:
- GTSRB
shufflenet - Federated Learning Round 5
Model Details
- Architecture: shufflenet
- Dataset: GTSRB
- Training Strategy: krum
- Round: 5
- Number of Classes: 43
Training Information
This model was trained using federated learning with the krum aggregation strategy.
Usage
import torch
model = torch.load("pytorch_model.bin")
model.eval()