nabil_oh/CnD_Classification
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pre trained 2 models

Browse Source
This commit is contained in:
Nabil Ould Hamou 2025-01-31 23:43:25 +01:00
parent 078dfa08eb
commit 8247fcf075
3 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
models/bayes_cat_dog_classifier.pth → models/bayes_128px.pth
View File

Binary file not shown.

BIN
models/bayes_256px.pth Normal file
View File

Binary file not shown.

2
train.py
View File

@ -50,7 +50,7 @@ def train_model(model_name, spinner):
dataset = TensorDataset(X_tensor, Y_tensor) dataset = TensorDataset(X_tensor, Y_tensor)
dataloader = DataLoader(dataset, batch_size=32, shuffle=True) dataloader = DataLoader(dataset, batch_size=32, shuffle=True)
model = CatDogClassifier() model = CatDogClassifier(img_size=IMG_SIZE)
model = model.to(DEVICE) model = model.to(DEVICE)
criterion = nn.CrossEntropyLoss() criterion = nn.CrossEntropyLoss()
optimizer = optim.Adam(model.parameters(), lr=0.001) optimizer = optim.Adam(model.parameters(), lr=0.001)