Change some absolute imports to relative

data/__init__.py

@@ -1,6 +1,6 @@
-from data.constants import *
+from .constants import *
-from data.config import resolve_data_config
+from .config import resolve_data_config
-from data.dataset import Dataset
+from .dataset import Dataset
-from data.transforms import *
+from .transforms import *
-from data.loader import create_loader
+from .loader import create_loader
-from data.mixup import mixup_target, FastCollateMixup
+from .mixup import mixup_target, FastCollateMixup

loss/__init__.py

@@ -1 +1 @@
-from loss.cross_entropy import LabelSmoothingCrossEntropy, SoftTargetCrossEntropy
+from .cross_entropy import LabelSmoothingCrossEntropy, SoftTargetCrossEntropy

models/__init__.py

@@ -1,4 +1,4 @@
-from models.model_factory import create_model
+from .model_factory import create_model
-from models.helpers import load_checkpoint, resume_checkpoint
+from .helpers import load_checkpoint, resume_checkpoint
-from models.test_time_pool import TestTimePoolHead, apply_test_time_pool
+from .test_time_pool import TestTimePoolHead, apply_test_time_pool
 

optim/__init__.py

@@ -1,3 +1,3 @@
-from optim.nadam import Nadam
+from .nadam import Nadam
-from optim.rmsprop_tf import RMSpropTF
+from .rmsprop_tf import RMSpropTF
-from optim.optim_factory import create_optimizer
+from .optim_factory import create_optimizer

scheduler/__init__.py

@@ -1,5 +1,5 @@
-from scheduler.cosine_lr import CosineLRScheduler
+from .cosine_lr import CosineLRScheduler
-from scheduler.plateau_lr import PlateauLRScheduler
+from .plateau_lr import PlateauLRScheduler
-from scheduler.step_lr import StepLRScheduler
+from .step_lr import StepLRScheduler
-from scheduler.tanh_lr import TanhLRScheduler
+from .tanh_lr import TanhLRScheduler
-from scheduler.scheduler_factory import create_scheduler
+from .scheduler_factory import create_scheduler