Official pytorch implementation of paper "Dual-Level Collaborative Transformer for Image Captioning" (AAAI 2021).
BSD 3-Clause "New" or "Revised" License
193
stars
31
forks
source link
Unable to load pretrained_model.pth #28
Open
yvfengZhong opened 2 years ago
When I try to execute the eval.py file, an error is reported “pretrained_model.pth is a zip archive(did you mean to use torch.jit.load()?)”.
Are you saving the model with torch 1.6 or above?
However, torch 1.1 is installed in the file environment.yml. Which one is right? Should I change to torch 1.6 or above?
I turned to execute the file train.py, and I spent a long time in this step. https://github.com/luo3300612/image-captioning-DLCT/blob/75a823d28cada1cd650258fc7ec86b3fa505d856/train.py#L248 I don't know if this is correct?
I use one GeForce RTX 3090Ti GPU with 24GB of memory.
I'm looking forward to your reply, thanks.