server_lib

Module Contents

`customDataset`
`LeNet`

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

device

load_data(config)

get_data(config)

class customDataset(root, transform=None)

Bases: torch.utils.data.Dataset

__getitem__(index)

__len__()

sample_return(root)

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)