Update

handetect/configs.py

@@ -2,7 +2,7 @@ import os
 import torch
 from torchvision import transforms
 from torch.utils.data import Dataset
-from models import *
+from handetect.models import *
 
 # Constants
 RANDOM_SEED = 123

handetect/eval.py

@@ -2,7 +2,7 @@ import os
 import torch
 from torchvision.transforms import transforms
 from sklearn.metrics import f1_score
-from models import * 
+from handetect.models import * 
 import pathlib
 from PIL import Image
 from torchmetrics import ConfusionMatrix

handetect/tuning.py

@@ -5,7 +5,7 @@ import torch.optim as optim
 from torchvision.transforms import transforms
 from torch.utils.data import DataLoader, random_split, Dataset
 from torchvision.datasets import ImageFolder
-from models import *
+from handetect.models import *
 from torch.utils.tensorboard import SummaryWriter #print to tensorboard
 from torchvision.utils import make_grid
 import optuna