yingkaisha/keras-unet-collection

"freeze_backbone" as RHS for both "freeze_backbone" and "freeze_batch_norm" arguments

ironhide23586 opened this issue · 1 comments

ironhide23586 commented

The same value of freeze_backbone is being erroneously fed in to both these arguments; check it out at the permalink below-

keras-unet-collection/keras_unet_collection/_model_transunet_2d.py

Line 430 in 6ee55b1

backbone=backbone, weights=weights, freeze_backbone=freeze_backbone, freeze_batch_norm=freeze_backbone, name=name)

Request a quick fix :)

Cheers! ψ(｀∇´)ψ

yingkaisha commented

Hi I've fixed the issue; it is available in pip install keras-unet-collection==0.1.7.