Federated-Learning-PyTorch
Federated-Learning-PyTorch copied to clipboard
Question
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)