webart
/
beerds
1
0
Fork 0
Code Issues 9 Pull Requests Actions Packages Projects Releases Wiki Activity

linter

This commit is contained in:
artem 2025-04-14 21:23:26 +03:00
parent 54642a15f5
commit d2b72ca916
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ml/cats.py
View File

@ -1,12 +1,13 @@
import json
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
import torch.nn as nn
from torchvision.datasets import ImageFolder # type: ignore
from torch.utils.data import DataLoader
from train import get_labels, load_model, get_loaders, train, show, DEVICE
from train import get_labels, load_model, get_loaders, train, show, DEVICE # type: ignore
ImageFile.LOAD_TRUNCATED_IMAGES = True
print(f"Using device: {DEVICE}")
IMG_SIZE = (180, 180)

5
ml/dogs.py
View File

@ -1,12 +1,13 @@
import json
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
import torch.nn as nn
from torchvision.datasets import ImageFolder # type: ignore
from torch.utils.data import DataLoader
from train import get_labels, load_model, get_loaders, train, show, DEVICE
from train import get_labels, load_model, get_loaders, train, show, DEVICE # type: ignore
ImageFile.LOAD_TRUNCATED_IMAGES = True
print(f"Using device: {DEVICE}")
IMG_SIZE = (180, 180)

2
ml/train.py
View File

@ -6,7 +6,7 @@ from torchvision.datasets import ImageFolder # type: ignore
from torch.utils.data import Dataset, DataLoader, random_split
from torchvision import transforms # type: ignore
import torchvision
from torchvision.models import ResNet50_Weights
from torchvision.models import ResNet50_Weights # type: ignore
from typing import Tuple
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")