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

training.models: fix 9b66867c

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-27 12:40:56 +01:00
parent 439ca350dd
commit 7c3aeda65e
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -334,7 +334,7 @@ def vit_resnet50_unet(num_patches,
transformer_mlp_head_units = [128, 64] transformer_mlp_head_units = [128, 64]
inputs = Input(shape=(input_height, input_width, 3)) inputs = Input(shape=(input_height, input_width, 3))
features = resnet50(inputs, weight_decay=weight_decay, pretraining=pretraining) features = list(resnet50(inputs, weight_decay=weight_decay, pretraining=pretraining))
features[-1] = transformer_block(features[-1], features[-1] = transformer_block(features[-1],
num_patches, num_patches,