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

finetune problems #20

Open successhaha opened 3 years ago

successhaha commented 3 years ago

when I finetune the pretrain model,there is an error. AssertionError File"trainval.py",line112, in trainval savedir_base=savedir_base /src/models/finetuning.py,line 66 in initassert(best_accuracy >0.1) can you help me ?

prlz77 commented 3 years ago

Hi, thanks for the interest in our code :)

This error means that the finetuner didn't find any pretrained model to load. Make sure you are pointing to the correct place in here: https://github.com/ElementAI/embedding-propagation/blob/master/exp_configs/finetune_exps.py#L86

If this does not resolve your problem, could you provide more information on what command you run, which exp_config, and what is the directory tree where you have the pre-trained weights?