Giters

AshwinRJ / Federated-Learning-PyTorch

Implementation of Communication-Efficient Learning of Deep Networks from Decentralized Data

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A small issue in creating an MLP model

mmalekzadeh opened this issue · comments

commented

It seems to me that the following line shouldn't be inside the loop and it should be moved outside the loop.

Federated-Learning-PyTorch/src/federated_main.py

Line 55 in 1bffc34

global_model = MLP(dim_in=len_in, dim_hidden=64,