Giters

nabsabraham / focal-tversky-unet

This repo contains the code for our paper "A novel focal Tversky loss function and improved Attention U-Net for lesion segmentation" accepted at IEEE ISBI 2019.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Batch_Size

AyeshaBhat opened this issue · comments

commented

Please tell me whether you have used a batch size of 16 or 8 with ISIC 2018 datatset
Because in paper you have mentioned that you are using batch size of 8 for isic dataset but in code you are using batch size of 16

commented

Please tell me how to fix this error

ValueError: Dimensions must be equal, but are 24 and 192 for '{{node mul_3}} = Mul[T=DT_FLOAT](Round_1, Round)' with input shapes: [?,24,32,1], [?,192,256,1].