Closed schmitzmatheus closed 1 year ago
On main.py (https://github.com/Sense-X/UniFormer/blob/main/image_classification/main.py#L276), we try to access key pos_embed on the loaded checkpoint, but for the checkpoints provided, the key is missing and so the code fails.
pos_embed
This happens when running run.sh (https://github.com/Sense-X/UniFormer/blob/main/image_classification/exp/uniformer_small/run.sh) with the provided checkpoints for fineturning (--finetune ).
I just saw #88
schmitzmatheus
commented
1 year ago schmitzmatheus
commented
1 year ago
On main.py (https://github.com/Sense-X/UniFormer/blob/main/image_classification/main.py#L276), we try to access key
pos_embedon the loaded checkpoint, but for the checkpoints provided, the key is missing and so the code fails.This happens when running run.sh (https://github.com/Sense-X/UniFormer/blob/main/image_classification/exp/uniformer_small/run.sh) with the provided checkpoints for fineturning (--finetune).