From 53252a59c6619bbf0d164a8c7fc6c98449b208ec Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Sun, 8 Feb 2026 01:09:40 +0100 Subject: [PATCH] training.models: fix glitch introduced in 3a73ccca --- src/eynollah/training/models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/eynollah/training/models.py b/src/eynollah/training/models.py index b0ad51c..5b23ecd 100644 --- a/src/eynollah/training/models.py +++ b/src/eynollah/training/models.py @@ -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 *