Rajarshi Bhowmik

icon indicating a website link https://kingsaint.github.io icon indicating an email [email protected]

icon company Rutgers University-New Brunswick icon location New York City icon location AI/NLP Research Engineer at Bloomberg L.P.

Results 4 comments of Rajarshi Bhowmik

ELQ failing at faiss index search while training on WebQSP dataset

Looks like the issue is with the `embedding_ctxt` . I assume it should be a 2D tensor/numpy array, but it is a 4D tensor when it is passed to `search_knn`.

CUDA error: device-side assert triggered

I am facing the same issue. Is it resolved? If yes, please let me know how.

CUDA error: device-side assert triggered

@ines-chami Looks like the `multi_c` option should be on? The following command worked. `python run.py --dataset YAGO3-10 --model AttH --max_epochs 500 --patience 10 --rank 200 --neg_sample_size -1 learning_rate 0.0005 --multi_c`...

CUDA error: device-side assert triggered

I used your command and got this error Traceback (most recent call last): File "run.py", line 191, in train(parser.parse_args()) File "run.py", line 142, in train train_loss = optimizer.epoch(train_examples) File "/common/home/rb897/KGEmb/optimizers/kg_optimizer.py",...