mobach626

模型训练问题

6

大佬，我用您的代码里面默认的swin-l pretrain用的coco和vg，但是训练好的模型看啥都是星空和棒球手套，所以想了解下您的训练细节，另外论文里面我看是用的swin-b训练的这里提供的权重实际上是l的对吧，下面是我的配置 train_file: [ 'datasets/train/coco_train_rmcocodev_ram.json', 'datasets/train/vg_ram.json', ] image_path_root: "/data/mobach" # size of vit model; base or large vit: 'swin_l' vit_grad_ckpt: False vit_ckpt_layer: 0 image_size: 224 batch_size: 20 # optimizer weight_decay:...