Improve imports quality by editting __init__.py

src/data/__init__.py

@@ -1 +1 @@
-from data import *
+from data.dataset import *

src/evaluate.py

@@ -11,11 +11,9 @@
 import torch
 from torch.utils.data import random_split
 
-from data.dataset import MiniFlickrDataset
-from model.loops import evaluate_dataset
-from model.model import Net
-from utils.config import Config
-from utils.load_ckp import download_weights
+from data import MiniFlickrDataset
+from model import evaluate_dataset, Net 
+from utils import Config, download_weights
 
 config = Config()
 parser = argparse.ArgumentParser()

src/model/__init__.py

@@ -0,0 +1,2 @@
+from model.loops import *
+from model.model import *

src/predict.py

@@ -12,9 +12,8 @@
 
 import torch
 
-from model.model import Net
-from utils.load_ckp import download_weights
-from utils.config import Config
+from model import Net
+from utils import Config, download_weights
 
 config = Config()
 parser = argparse.ArgumentParser()

src/training.py

@@ -13,12 +13,9 @@
 import torch.optim as optim
 from torch.utils.data import random_split
 
-from data.dataset import MiniFlickrDataset, get_loader
-from model.model import Net
-from model.loops import train_epoch, valid_epoch, test_step
-from utils.config import Config
-from utils.load_ckp import load_ckp
-from utils.lr_warmup import LRWarmup
+from data import MiniFlickrDataset, get_loader
+from model import Net, train_epoch, test_step, valid_epoch
+from utils import Config, load_ckp, LRWarmup
 
 config = Config()
 parser = argparse.ArgumentParser()

src/utils/__init__.py

@@ -1 +1,3 @@
-from utils import *
+from utils.config import *
+from utils.load_ckp import *
+from utils.lr_warmup import *