Yuzmi commited on
Commit
4de61ef
1 Parent(s): 8ef6259

Delete Python

Browse files
Files changed (1) hide show
  1. Python +0 -51
Python DELETED
@@ -1,51 +0,0 @@
1
- import torch
2
- from torchvision import datasets, transforms
3
- import torch.nn as nn
4
- import torch.optim as optim
5
-
6
- # Define your model class
7
- class TatsukichiHayamaClassifier(nn.Module):
8
- def __init__(self, num_classes=10):
9
- super(TatsukichiHayamaClassifier, self).__init__()
10
- # Define your model layers here
11
- self.conv1 = nn.Conv2d(3, 64, kernel_size=3, stride=1, padding=1)
12
- self.relu = nn.ReLU()
13
- self.pool = nn.MaxPool2d(kernel_size=2, stride=2)
14
- self.fc1 = nn.Linear(64 * 64 * 64, num_classes) # Update input size based on your image dimensions
15
-
16
- def forward(self, x):
17
- x = self.conv1(x)
18
- x = self.relu(x)
19
- x = self.pool(x)
20
- x = x.view(-1, 64 * 64 * 64) # Update this size based on your image dimensions
21
- x = self.fc1(x)
22
- return x
23
-
24
- # Load dataset from PyTorch's ImageFolder
25
- # Adjust the 'root' parameter to the path where your dataset is stored
26
- train_dataset = datasets.ImageFolder(root="path/to/your/dataset", transform=transforms.ToTensor())
27
-
28
- # Create a DataLoader for training
29
- dataloader = torch.utils.data.DataLoader(train_dataset, batch_size=32, shuffle=True)
30
-
31
- # Create an instance of TatsukichiHayamaClassifier
32
- your_num_classes = 10 # Adjust this based on your dataset
33
- model = TatsukichiHayamaClassifier(num_classes=your_num_classes)
34
-
35
- # Model, criterion, and optimizer
36
- criterion = nn.CrossEntropyLoss()
37
- optimizer = optim.Adam(model.parameters(), lr=0.001)
38
-
39
- # Training loop
40
- num_epochs = 10
41
-
42
- for epoch in range(num_epochs):
43
- model.train()
44
- for images, labels in dataloader:
45
- optimizer.zero_grad()
46
- outputs = model(images)
47
- loss = criterion(outputs, labels)
48
- loss.backward()
49
- optimizer.step()
50
-
51
- print(f'Epoch {epoch+1}/{num_epochs}, Loss: {loss.item()}')