search

litian96 / FedProx

Federated Optimization in Heterogeneous Networks (MLSys '20)
MIT License
642 stars 157 forks source link

No model of "'mnist.cnn" #3

Closed cshjin closed 5 years ago

cshjin commented 5 years ago

Hi there, I do see you have the option of using CNN on the MNIST dataset. But I don't see the implementation in the model. Would you provide it later?

BTW, I was also on ICML this year, but was unable to attend the poster session. Would you put your poster on your homepage as well?

H.

litian96 commented 5 years ago

Thanks for your interest! You could use any CNN architecture for MNIST. The most relevant one is: https://github.com/TalwalkarLab/leaf/tree/master/models/femnist of the LEAF benchmark. I am sure you could successfully get CNN running on MNIST following this piece of code. The poster is up.

cshjin commented 5 years ago

Thanks. Issue closed.