search

ViTAE-Transformer / Remote-Sensing-RVSA

The official repo for [TGRS'22] "Advancing Plain Vision Transformer Towards Remote Sensing Foundation Model"
MIT License
417 stars 32 forks source link

Model pth #24

Open HuangShiqi128 opened 1 year ago

HuangShiqi128 commented 1 year ago

Hi,

May I ask what are vit_rvsa.pth, vitae_rvsa.pth, vit_rvsa_kvdiff.pth, vitae_rvsa_kvdiff.pth? At first, I thought the 'Model' column in each downstream task contains checkpoints for traning model on different dataset, but it turns out that they are the same for different tasks and datasets.

Thank you.

DotWang commented 1 year ago

@HuangShiqi128

At first, I thought the 'Model' column in each downstream task contains checkpoints for traning model on different dataset

You are right, they are the saved checkpoint after finetuning on different tasks.

HuangShiqi128 commented 1 year ago

OK, so is it true that I can directly run test using that pth file corresponding to different task and dataset and get results in paper?

DotWang commented 1 year ago

@HuangShiqi128 yes