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

Question

SamiurRahman1 opened this issue · comments

Hello, i am just curious to know, what does line 72 in update.py do? Does it forward the image to the model?
log_probs = model(images)