runtime erro about fl-bench

[KarhouTam]

Hi, @elegy112138.
Could you offer the runtime arguments that trigger this bug?

[elegy112138]

"I see that your code has been updated, and it's different from what I previously pulled.
my arguments is
--dataset cifar10 --visible 1 --global_epoch 200
server/fedavg
parser.add_argument(
"-m",
"--model",
type=str,
# default="ModifiedLeNet5",
default="mobile",
choices=["lenet5", "2nn", "avgcnn", "mobile", "res18", "alex", "sqz", "ModifiedLeNet5"],
)

[KarhouTam]

Sorry for the old version bug and you can try the latest if available.
According to your information, it seems that the bug raised due to the incompatibility bewteen model and dataset. You can check that part by yourself to find the bug.
If the latest version also raise that bug, please infer me.

[KarhouTam]

This issue is closed due to long time no response.