qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-02-20 16:32:03 +01:00
Code Issues Projects Releases Wiki Activity

training: use relative imports

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
src/eynollah/training/inference.py
View file

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

2
src/eynollah/training/models.py
View file

@ -30,7 +30,7 @@ from tensorflow.keras.layers import (
from tensorflow.keras.models import Model from tensorflow.keras.models import Model
from tensorflow.keras.regularizers import l2 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] ##mlp_head_units = [512, 256]#[2048, 1024]
###projection_dim = 64 ###projection_dim = 64

2
src/eynollah/training/weights_ensembling.py
View file

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