May 10, 2022
@Polydatum I think you are getting this error during training, it could be because of a mismatch on batch dimension. So, try squeezing 0th dimension with torch.squeeze.
Sign up to discover human stories that deepen your understanding of the world.
Free
Distraction-free reading. No ads.
Organize your knowledge with lists and highlights.
Tell your story. Find your audience.
Membership
Read member-only stories
Support writers you read most
Earn money for your writing
Listen to audio narrations
Read offline with the Medium app
Written by Ankan Sharma
Avid Learner | Areas of Interest are CV, NLP, and Graph DL | Python Backend Developer | Mobile Developer in Flutter | Gamer at Heart
