server_lib

Module Contents

`load_data`(config)
`get_data`(config)
`get_net`(config)
`train_model`(net, trainloader)
`test_model`(net, testloader)
`save_intial_model`(config)

device

load_data(config)

get_data(config)

class LeNet(in_channels=1, num_classes=10)

Bases: torch.nn.Module

forward(x)

get_net(config)

train_model(net, trainloader)

test_model(net, testloader)

save_intial_model(config)