UNITER icon indicating copy to clipboard operation
UNITER copied to clipboard

Research code for ECCV 2020 paper "UNITER: UNiversal Image-TExt Representation Learning"

Results 45 UNITER issues
Sort by recently updated
recently updated
newest added

Thanks for your great work! Could you please share the code for create img / txt db for CC / SBU ? Thanks again!

When I launched the 2nd stage training using : ``` horovodrun -np 4 python pretrain_vcr.py --config config/pretrain-vcr-base-4gpu.json \ --output_dir $PRETRAIN_VCR_EXP ``` An error occured, which is: ``` Traceback (most recent...

Hi, as title says, I do not find decoder in UniterModel. But the paper indicates Uniter has a decoder. Do you release the code of decoder?

when I run python train_re.py --config config/train-refcoco-base-1gpu.json --output_dir $RE_EXP I just get this result. 03/11/2021 11:44:06 - INFO - __main__ - Loading Train Dataset /txt_db/refcoco_train.db, /img_db/re_coco_gt /img_db/re_coco_gt /txt_db/refcoco_train.db Traceback (most...

WARNING: Detected NVIDIA NVIDIA A100-PCIE-40GB GPU, which is not yet supported in this version of the container How to deal with it ?

How to measure the loss weight of different pre-training tasks? Which task's loss determines the model training convergence?

**This is more of a question than issue.** In model.py at line number 330 gather index has been used to re-orient text and image embedding after concatenation. I am trying...

Hi, I met the following problem recently: W /tmp/pip-install-fzrlm1c4/horovod/horovod/common/stall_inspector.cc:105] One or more tensors were sub mitted to be reduced, gathered or broadcasted by subset of ranks and are waiting for...

how to generate image feature from ground true? the code just provide the stript that generates the image from detected bbx instead of ground true.

I encountered the following error on GPU 3090, but it can run successfully on Titan Xp. I suspect that the Cuda version in the container is relatively low. So has...