search

mlvlab / SpeaQ

Official PyTorch implementation of "Groupwise Query Specialization and Quality-Aware Multi-Assignment for Transformer-based Visual Relationship Detection" (CVPR 2024).
24 stars 2 forks source link

Setup Installation #4

Closed Mikael17125 closed 2 months ago

Mikael17125 commented 3 months ago

Hi I have faced some issues, can you help me?

Traceback (most recent call last): File "/home/mikael/Code/SpeaQ-main/train_iterative_model.py", line 14, in from SpeaQ.engine import JointTransformerTrainer ModuleNotFoundError: No module named 'SpeaQ'

I just follow the installation on your README.md and run the code

jonghakim35 commented 3 months ago

The code should work after modifying the code directory from /home/mikael/Code/SpeaQ-main to /home/mikael/Code/SpeaQ.