search

BUAADreamer / SPN4CIR

[ACM MM 2024] Improving Composed Image Retrieval via Contrastive Learning with Scaling Positives and Negatives
https://arxiv.org/abs/2404.11317
MIT License
11 stars 0 forks source link

Training inexplicably interrupted #2

Closed yanglichang2849 closed 2 weeks ago

yanglichang2849 commented 1 month ago

Dear author,I've encountered a bug while running blip4cir/train.py. The training process terminates unexpectedly during the encoding of bank features without any error messages. Could you please help me with this issue? Thank you:) Screenshot 2024-09-20 202659

BUAADreamer commented 1 month ago

其他的模型您可以跑吗

yanglichang2849 commented 2 weeks ago

CLIP4CIR可正常跑，但BLIP模型异常中断

BUAADreamer commented 2 weeks ago

If you need to run the blip4cir and blip24cir training, try to prepare a machine with sufficient cuda memory (at least A100 40G) and RAM (300G)

By the way, since the size of the bank is too large, there is no plan to put the bank directly on the huggingface, sorry about this