Comments (4)
I am not sure what the exact problem is, replacing mini with custom dataset should work. It seems the batch_size of logits and label did not match, could you check their shape?
from prototypical-network-pytorch.
yes, the shapes are different ..
Shape of logits torch.Size([5, 5])
Shape of label torch.Size([75])
Are there any constraints for the custom dataset for the code to run? Should I tweak any parameters accordingly?
I used the following parameters
shot=5
query=15
train way=30
test way=5
I have used mini during training and custom dataset for validation which has 17 classes and 6 images per class.
from prototypical-network-pytorch.
I referred to the paper and modified query=3 and shot=3 as per my custom training support. Now the code is working fine.
Thanks for the quick response! :)
from prototypical-network-pytorch.
I remember there should not be constraints for custom dataset. You may check the intermediate shapes to see what the exact problem is. The shape of logits should be [n_way * n_query, n_way], and shape of label should be [n_way * n_query].
from prototypical-network-pytorch.
Related Issues (18)
- How to get the mini-imageNet dataset? HOT 2
- Single image prediction HOT 1
- How train with a new dataset HOT 1
- Selecting support and query sets randomly HOT 1
- 训练、测试曲线图training graph,testing graph HOT 2
- confusion matrix plot for evaluation
- Dataset could not be downloaded. HOT 1
- Cannot get the reported results. HOT 1
- Problem about labels
- About loss function HOT 13
- No of epochs to be trained for achieving the given results HOT 2
- Can you provide demo.py? HOT 1
- about the memory HOT 5
- Change step-size and max-epochs to get higher performance HOT 1
- LICENSE? HOT 1
- Running on Google Colab HOT 1
- DataSets 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 prototypical-network-pytorch.