training: use relative imports

This commit is contained in:
Robert Sachunsky 2026-02-08 01:11:44 +01:00
parent 7b7ef041ec
commit 37338049af
3 changed files with 3 additions and 3 deletions

View file

@ -25,7 +25,7 @@ from .gt_gen_utils import (
resize_image,
update_list_and_return_first_with_length_bigger_than_one
)
from .models import (
from ..patch_encoder import (
PatchEncoder,
Patches
)

View file

@ -30,7 +30,7 @@ from tensorflow.keras.layers import (
from tensorflow.keras.models import Model
from tensorflow.keras.regularizers import l2
from eynollah.patch_encoder import Patches, PatchEncoder
from ..patch_encoder import Patches, PatchEncoder
##mlp_head_units = [512, 256]#[2048, 1024]
###projection_dim = 64

View file

@ -12,7 +12,7 @@ tf_disable_interactive_logs()
import tensorflow as tf
from tensorflow.keras.models import load_model
from .models import (
from ..patch_encoder import (
PatchEncoder,
Patches,
)