visinf / cos-cvae Goto Github PK

Hi,

Thank you for your source code!

I wonder what the role of eval_diversity.py is, since there are no diversity metrics, like novel words, new sentences in the file. So what is the difference between eval.py and eval_diversity.py?

Thank you
Ryan.

Hi,

Thank you for your code! It seems the function of beam search does not end the sentence when it meets the token, causing the sentences by beam search (e.g. BS=5) not to be different from each other. Could you check it again? And by the way, could you add DBS in the reference stage?

Thank you

I'm sorry to bother you, but I can not find the code for Unique and Novel, can U show me where the code is? Thank U so much

Thanks in advance for your reply

Traceback (most recent call last):
File "train_noc.py", line 186, in
for data in train_bar:
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/tqdm/std.py", line 1178, in iter
for obj in iterable:
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1203, in _next_data
return self._process_data(data)
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 1229, in _process_data
data.reraise()
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/torch/_utils.py", line 434, in reraise
raise exception
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/tuxiangsuo1502/miniconda3/envs/hjl/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/data/txs/hjl/coscvae/modules/data_loader_noc.py", line 263, in getitem
caps = self.inv_annotations[str(image_idx)]
KeyError: "b'262146'"

How can it run normally? Thank you in advance for your reply