Delete Python
Browse files
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()}')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|