告密者的下场（2/5）

out = (out)

out = (out)

out = (out)

out = (batchsize, -1)

out = (out)

out = (out, dim=1)

return out

class TrainingDataSet(Dataset):

def __init__(self):

super(TrainingDataSet, self).__init__()

 = X_train

 = y_train

def __getitem__(self, index):

t = [index, 0:36]

t = (t).view(6, 6)

return t, [index]

def __len__(self):

return len()

class TestDataSet(Dataset):

def __init__(self):

super(TestDataSet, self).__init__()

 = X_validate

 = y_validate

def __getitem__(self, index):

t = [index, 0:36]

t = (t).view(6, 6)

return t, [index]

def __len__(self):

return len()

def cnn_classification():

batch_size = 256

trainDataLoader = DataLoader(TrainingDataSet(), batch_size=batch_size, shuffle=False)

testDataLoader = DataLoader(TestDataSet(), batch_size=batch_size, shuffle=False)

epoch_num = 200

#lr = 

lr = 

net = VGGBaseSimpleS2().to(device)

print(net)

# loss