search

ServiceNow / embedding-propagation

Codebase for Embedding Propagation: Smoother Manifold for Few-Shot Classification. This is a ServiceNow Research project that was started at Element AI.
Apache License 2.0
208 stars 21 forks source link

train error #16

Closed successhaha closed 3 years ago

successhaha commented 3 years ago

hi,when I pretrian the model on CUB_200_2011,there is an error.How to solve it? Thank you. File "/home/xww/embedding-propagation-master/embedding_propagation/batch_embedding_propagation.py", line 14, in forward return embedding_propagation(x, self.alpha, self.rbf_scale, self.norm_prop, propagator=propagator) File "/home/xww/embedding-propagation-master/embedding_propagation/batch_embedding_propagation.py", line 57, in embedding_propagation weights = get_similarity_matrix(x, rbf_scale) File "/home/xww/embedding-propagation-master/embedding_propagation/batch_embedding_propagation.py", line 45, in get_similarity_matrix b, e, c = x.size() ValueError: not enough values to unpack (expected 3, got 2)

prlz77 commented 3 years ago

Fixed! This is a recent bug due to the introduction of batch_ep. Could you pull and try again?

Thanks!