/project
    ㄴdata
	  ㄴdataset_name
			ㄴtrain_data
				ㄴclass1
				ㄴclass2
				ㄴclass3
			ㄴtest_data
				ㄴclass1
				ㄴclass2
				ㄴclass3
    ㄴImageFolder_EX.py

trans = transforms.Compose([
            transforms.ToTensor()
])

train_data = torchvision.datasets.ImageFolder(root='data/custom_data/train_data', transform=trans)

data_loader = DataLoader(dataset=train_data, batch_size=8, shuffle=True)

test_data = torchvision.datasets.ImageFolder(root='data/custom_data/test_data', transform=trans)

test_loader = DataLoader(dataset=test_data, batch_size=len(test_data))

class CNN(nn.Module):
    def __init__(self):
        super(CNN, self).__init__()
        self.layer1 = nn.Sequential(
            nn.Conv2d(3, 6, 5),
            nn.ReLU(),
            nn.MaxPool2d(2),
        )
		.
		.
		.
		# Full Code 참조