Reorganize import statements for consistency and clarity in model and training scripts

model.py

@@ -1,10 +1,16 @@
-import torch
-import torch.nn as nn
-import torch.nn.functional as F
-from age_encoder import AgeSinusoidalEncoder, AgeMLPEncoder
-from backbones import Block
-from typing import Optional, List
 import numpy as np
+from typing import Optional, List
+from backbones import Block
+from age_encoder import AgeSinusoidalEncoder, AgeMLPEncoder
+import torch.nn.functional as F
+import torch.nn as nn
+import torch
+import sys
+from pathlib import Path
+
+_PROJECT_ROOT = Path(__file__).resolve().parent
+if str(_PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(_PROJECT_ROOT))
 
 
 class TabularEncoder(nn.Module):