YubeiZheng

RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

6

when i want to trian RelTR on Open Images V6 with a single GPU python main.py --dataset oi --img_folder /home/ybz/RelTR/data/oi/images/ --ann_path /home/ybz/RelTR/data/ --batch_size 1 --output_dir ckpt1 

Results visualization

2

Hello, I would like to ask how The Visualization of prediction shitting the ground truth from VidVRD and TRACE in your article is implemented?

Train on Custom Images

2

Thank you very much for sharing, and I want to train my own images but don't know how to annotate relations. Could you please tell me what tools to use...

## ❓ Questions and Help I want to use my own dataset but the data has no attributes, how to turn off the attribute head in both detector pretraining part...