Comments (1)
I have solved this problem,
“Dimension out of range (expected to be in range of [-2, 1], but got 2)”
The reason for this issue is that scores was originally a three-dimensional tensor, but after the operation
scores = pack_padded_sequence(scores, decode_lengths, batch_first=True)[0],
it became a two-dimensional vector.
Here are two solutions:
1.Use the pad_packed_sequence function to restore it.
2.Before reducing dimensionality, use unpacked_scores = scores, then: _, preds = torch.max(unpacked_scores, dim=2).
Note: The original code used scores_copy, I replaced it to avoid errors.
from a-pytorch-tutorial-to-image-captioning.
Related Issues (20)
- Asking about List of Packages Version
- Failed to create .hdf5 files
- bleu-4 does not increase HOT 2
- Code stopped @ train.py with this error,
- Anyone , Please help to solve this error HOT 3
- If i want to use RL for this , how will i do ?
- eval.val is not working.
- Dimension error HOT 2
- Dataset not available HOT 3
- ValueError: max() arg is an empty sequence HOT 3
- High loss and low bleu-4 for training HOT 7
- Example Notebook?? HOT 1
- In eval.py, it seems that for each image, the model has to recalculate 5 times, is it too inefficient?
- Please help me with the error:RuntimeError: Expected target size [32, 9490], got [32, 51] HOT 1
- I think this a bug. caption.py 140 HOT 5
- RuntimeError: CUDA error: device-side assert triggered
- Only the author mentions half the bleu-4 score HOT 2
- can this model detect and recognize text in images containing text HOT 1
- The code is wrong..... HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from a-pytorch-tutorial-to-image-captioning.