rmokady / CLIP_prefix_caption

Simple image captioning model
MIT License
1.29k stars 214 forks source link

it would be None #54

Open enes3774 opened 2 years ago

enes3774 commented 2 years ago

https://github.com/rmokady/CLIP_prefix_caption/blob/1ad805a844a62ab2e5480479aa021bccf0d4d12a/train.py#L230 I think if your labels are None you can create labels data. So it would be if labels is None