JeremyXSC / DMRL

MIT License
1 stars 0 forks source link

配置文件设置 #1

Open SweetLittleLies opened 1 month ago

SweetLittleLies commented 1 month ago

您好,非常感谢您所提供的文章和代码。在读了您的文章之后我下载了您的代码。想实现下文章中所提到的在finegpr上预训练然后在常用的数据集上进行微调。在代码中的一些配置选择不太清楚。想请问下您是怎么设置的。 如:在train.py中使用的是make_dataloader_cross(cfg)还是 make_dataloader(cfg)呢?model = make_model_cross还是选择model = make_model。 在make_model.py中选择cfg.MODEL.NAME == 'transformer'还是用model = Backbone?且在这里的预训练加载默认的是pretrain_choice == 'imagenet'。我改成了pretrain_choice == 'finetune',self.load_param(model_path),然后把defaults.py文件中的MODEL.PRETRAIN_PATH换成您在github所说的预训练文件这样设置是对的吗?因为我看def load_param(self, trained_path):和def load_param_finetune(self, model_path):似乎默认没有调用。这样做加载了finegpr的预训练权重。但是def load_param_finetune这里的微调的预训练加载还是不清楚在哪里进行了使用。 非常抱歉问了好多问题,给您添麻烦了。如果您有时间的话能否为我解答一二。再次感谢您所做的工作。

JeremyXSC commented 1 week ago

你好,感谢你对我们工作的关注;请将具体的训练日志发送到我的邮箱吧,我们邮箱沟通,谢谢!