我相信学习深度学习的同学应该遇到过这样一个案例,我呢就是简单的改了一个RESNET50变为RESNET18。
为此我按照GPT提供的思路久久不能解决问题。今天终于在师兄的帮助下,搞明白了。对一个网络,我们改网路的师兄,一定要注意修改预训练的数据。往往在之前那个网络有预训练的数据,我们一定要取消读取该预训练的数据。 #restore_model(config["training.pretrained_checkpoint_path"] # self.backbone, self.decoder, self.optimizer, # logger=logger) 我将restore_model 注释了就好了
修改网络之后的注意事项(一)
最新推荐文章于 2024-09-13 19:41:31 发布