ShenZheng2000
ShenZheng2000
### What's the feature? We can perform inference using Stable Diffusion, but it cannot be used for training or testing. Please implement this functionality. ### Any other context? _No response_
### Describe the feature I appreciate the implementation of `ImageCaptionInferencer `. Could you also implement training functionality for image captioning? ### Will you implement it? - [ ] I would...
Hello, authors! I have diffculty understanding this [line](https://github.com/fnzhan/MoNCE/blob/1a666e26f4312db6867db7ba754e4f6901c6fde5/CUT_MoNCE/models/sinkhorn.py#L33) of code. ``` C = torch.einsum('bid,bod->bio', q, k) ``` What does torch.einsum do here? Could you provide a simple example? Thanks!
Hello, authors. I really appreciate your nice work on image draining. I would like to know how long it took to converge your model. Are you using a single GPU...
Hi, authors! I am writing a survey paper on image deraining and am very interested in your BID dataset. Could you tell me how many training and testing images are...
How many GPUs do you use for training, and how many hours does it take to converge the model?
Hello, authors! Could you release the pretrained model so I can reproduce your results? Thanks!
The paper demonstrates the significant influence of two-stage consistency on mAP. However, in the code, the "CONSISTENCY: True" configuration parameter is not utilized. Moreover, I am curious about the performance...
How can we visualize **teachers' prediction**, **student's prediction guided by the teacher**, and **teachers' prediction for consisteny loss** to achieve results similar to those in Fig. 3 of the paper?...
Hello, authors! Thanks for your excellent work. I have trouble with multi-GPU training. My command line looks like this: ``` python train.py --dataroot $dataset_path--name $model_name--gpu 0,1,2,3 --batch_size 1 ``` And...