diff --git a/examples/mt-pytorch-callable/client.py b/examples/mt-pytorch-callable/client.py index 05e4757fe141..1d8f35fb1ad5 100644 --- a/examples/mt-pytorch-callable/client.py +++ b/examples/mt-pytorch-callable/client.py @@ -1,20 +1,20 @@ +import datetime import os import time -import datetime import warnings from collections import OrderedDict +import flwr as fl import tensorflow as tf import torch import torch.nn as nn import torch.nn.functional as F +from flwr.common.serde import client_message_from_proto, server_message_from_proto from torch.utils.data import DataLoader from torchvision.datasets import CIFAR10 from torchvision.transforms import Compose, Normalize, ToTensor from tqdm import tqdm -import flwr as fl -from flwr.common.serde import client_message_from_proto, server_message_from_proto # ############################################################################# # 1. Regular PyTorch pipeline: nn.Module, train, test, and DataLoader