From ae3b6916ee2d49d36514ae5bc308886fb47d0d51 Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Sun, 1 Mar 2026 18:39:30 +0100 Subject: [PATCH] assert within vit_resnet50_unet model is commented out since arising assert error --- src/eynollah/training/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/eynollah/training/models.py b/src/eynollah/training/models.py index a1d52ce..201dc4c 100644 --- a/src/eynollah/training/models.py +++ b/src/eynollah/training/models.py @@ -424,7 +424,7 @@ def vit_resnet50_unet(n_classes, patch_size_x, patch_size_y, num_patches, mlp_he # Skip connection 2. encoded_patches = Add()([x3, x2]) - assert isinstance(x, Layer) + #assert isinstance(x, Layer) encoded_patches = tf.reshape(encoded_patches, [-1, x.shape[1], x.shape[2] , int( projection_dim / (patch_size_x * patch_size_y) )]) v1024_2048 = Conv2D( 1024 , (1, 1), padding='same', data_format=IMAGE_ORDERING,kernel_regularizer=l2(weight_decay))(encoded_patches)