From 86adaf299ade201b178fe851c7b4f884a680fc0c Mon Sep 17 00:00:00 2001
From: Robert Sachunsky <sachunsky@informatik.uni-leipzig.de>
Date: Tue, 19 May 2026 03:17:31 +0200
Subject: [PATCH] =?UTF-8?q?training.models.transformer=5Fblock:=20tf.resha?=
 =?UTF-8?q?pe=20=E2=86=92=20Keras=20Reshape=20layer?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/eynollah/training/models.py             | 9 ++++-----
 src/eynollah/training/reload-models-v0.8.mk | 7 ++++---
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/eynollah/training/models.py b/src/eynollah/training/models.py
index 3494249..f700d14 100644
--- a/src/eynollah/training/models.py
+++ b/src/eynollah/training/models.py
@@ -309,11 +309,10 @@ def transformer_block(img,
         # Skip connection 2.
         encoded_patches = Add()([x3, x2])
 
-    encoded_patches = tf.reshape(encoded_patches,
-                                 [-1,
-                                  img.shape[1],
-                                  img.shape[2],
-                                  projection_dim // (patchsize_x * patchsize_y)])
+    encoded_patches = Reshape(target_shape=(img.shape[1],
+                                            img.shape[2],
+                                            projection_dim // (patchsize_x * patchsize_y)),
+                              name="reshape_patches")(encoded_patches)
     return encoded_patches
 
 def vit_resnet50_unet(num_patches,
diff --git a/src/eynollah/training/reload-models-v0.8.mk b/src/eynollah/training/reload-models-v0.8.mk
index b7a38dd..07be7cf 100644
--- a/src/eynollah/training/reload-models-v0.8.mk
+++ b/src/eynollah/training/reload-models-v0.8.mk
@@ -26,16 +26,17 @@ RELOADABLE_MODELS = \
 all: $(RELOADABLE_MODELS)
 
 $(MODELS_DST)/%: $(MODELS_SRC)/%
-	mkdir -p $@
 	test -e $</config.json || exit 1
-	eynollah-training train --force \
+	{ mkdir -p $@ \
+	&& eynollah-training train --force \
 		with $</config.json \
 		reload_weights=True \
 		continue_training=False \
 		dir_output=$(dir $@) \
 		dir_of_start_model=$< \
+	&& cp $</config.json $@/config.json \
+	|| { rm -rf $@; false; }; } \
 	2>&1 | tee $(notdir $<).log
-	cp $</config.json $@/config.json
 
 compare: 
 	for i in `find $(MODELS_DST) -mindepth 2`;do \