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.models: fix glitch introduced in 3a73ccca

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-08 01:09:40 +01:00
parent ea285124ce
commit 53252a59c6
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -443,7 +443,6 @@ def vit_resnet50_unet(num_patches,
# Skip connection 2.
encoded_patches = Add()([x3, x2])
assert isinstance(x, Layer)
encoded_patches = tf.reshape(encoded_patches,
[-1, x.shape[1], x.shape[2],
transformer_projection_dim // (transformer_patchsize_x *