shiyuanh / tane Goto Github PK

Hey there! Will you release the code soon? Thanks!

Great work!
I would like to be able to cite your paper in my own work. However, I have encountered some confusion while reading your code.

1. Equation 6 in the paper does not seem to appear in the code, but instead is a calibration module that also uses the attention mechanism.
2. Using the example of training command for 5-way 1-shot FSOR does not seem to yield the correct ATT, ATT-G or SEMAN-G results for FSOR. According to the description in the bottom left corner of page 4 of the paper, the agg parameter should be set to mlp instead of avg.
3. The current code seems to support only a single negative class estimate.

Looking forward to your reply, best wishes!

Hi,

I am highly interested in your work, TANE, and want to reproduce the results.
I found out that your links for pre-trained models only include 'archive' directory with some data not the pre-trained model.
Could you check this problem?

Best