dino-paddle

Dino paddle implemetion

full_checkpoint 提取码：uyt7

linear_checkpoint 提取码：3j5t

使用

训练backbone

在train_vit.sh设置如下参数：

• --gpus：gpu ids
• --data_path：训练集路径
• --output_dir：日志、模型保存的目录

训练分类层

在train_linear.sh设置如下参数：

• --gpus：gpu ids

• --data_path：训练集路径

• --pretrained_weights：backbone的预训练模型。默认即可

• --output_dir：日志、模型的保存目录

• --batch_size：32，训练分类层、评估模型时都是32。四卡的话就是4*32=128

评估模型

在eval.sh中设置如下参数：

• --data_path：验证集路径
• --pretrained_weights：backbone的预训练模型
• --pretrained_linear：分类层的预训练模型