search

farrell236 / RATCHET

RAdiological Text Captioning for Human Examined Thoraxes
MIT License
38 stars 9 forks source link

DenseNet121 Pre-trained Weights Error #5

Open katelyn98 opened 3 months ago

katelyn98 commented 3 months ago

I am using the pre-trained weights in the classifer/assets/ folder and I get the following error when running the code:

ValueError: The name "zero_padding2d" is used 2 times in the model. All layer names should be unique.

Any help would be appreciated.

farrell236 commented 3 months ago

Hi Katelyn, the pre-trained weights in classifier/assets folder is for the classifier model only, they are not for DenseNet121.

The model weights for RATCHET is located in the main README.md file.

It's also ok to ignore the warning message No Pretrained DenseNet-121 weights specified when restoring RATCHET weights. This was debug message for training if wanting to start from a pretrained densenet (e.g. a densenet trained on chexpert, etc...).

Hope this helps!