KarhouTam / FL-bench

Benchmark of federated learning. Dedicated to the community. 🤗

GNU General Public License v3.0

503 stars 82 forks source link

runtime erro #47

Closed elegy112138 closed 11 months ago

elegy112138 commented 12 months ago

"I want to specify the model as MobileNetV2 using your code, but I keep getting an error, 'RuntimeError: size mismatch (got input: [10], target: [32]).' Do you know what's happening?"

KarhouTam commented 12 months ago

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

elegy112138 commented 12 months ago

"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 commented 12 months ago

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 commented 11 months ago

This issue is closed due to long time no response.