VFA icon indicating copy to clipboard operation
VFA copied to clipboard

Published 20 hours ago • verified publisher icon csuhan

Error occurs when I use command to fine-tune voc data

Open luoyq6 opened this issue 2 years ago • 7 comments

Hello, an error occurs when I use the command to fine-tune voc data, why is this happening?

image

image

luoyq6 avatar Mar 21 '23 11:03 luoyq6

Do you place the base-training checkpoint here? https://github.com/csuhan/VFA/blob/e35411eb22b4fc48b524debe58dc7c09be2bf9a6/configs/vfa/voc/vfa_split1/vfa_r101_c4_8xb4_voc-split1_10shot-fine-tuning.py#L27

You can re-download it and try again.

csuhan avatar Mar 21 '23 17:03 csuhan

你把基地训练检查站放在这里吗？

https://github.com/csuhan/VFA/blob/e35411eb22b4fc48b524debe58dc7c09be2bf9a6/configs/vfa/voc/vfa_split1/vfa_r101_c4_8xb4_voc-split1_10shot-fine-tuning.py#L27

您可以重新下载它，然后重试。 我是下载的你的模型权重，并重新下载了，但是还是出现这种情况 image image

luoyq6 avatar Mar 22 '23 01:03 luoyq6

检查一下pytorch是否能加载model： model = torch.load('work_dirs/vfa_r101_c4_8xb4_voc-split1_base-training/iter_18000.pth',map_location='cpu')

csuhan avatar Mar 22 '23 06:03 csuhan

检查一下pytorch是否能加载model： model = torch.load('work_dirs/vfa_r101_c4_8xb4_voc-split1_base-training/iter_18000.pth',map_location='cpu')

你好，打扰了，我运行以下脚本，出现这个错误，是否是没加载模型？ image image

luoyq6 avatar Mar 22 '23 06:03 luoyq6

可能是模型权重参数在下载时有问题或者您这边提供的模型权重参数有问题，我已经重新下载了还是出现错误，但是我用我自己训练好的模型去微调，不会出现这个错误。 image

luoyq6 avatar Mar 22 '23 06:03 luoyq6

You can try with newer Pytorch. I tested the pretrained checkpoint on Pytorch 1.12 and it still works.

csuhan avatar Mar 22 '23 10:03 csuhan

您可以尝试使用较新的 Pytorch。我在 Pytorch 1.12 上测试了预训练的检查点，它仍然有效。 你好，我已经可以用你的权重微调了，他可能是由于文件命名的问题，将vfa_r101_c4_8xb4_voc-split1_base-training_iter_18000.pth修改为iter_18000.pth就可以工作

luoyq6 avatar Mar 23 '23 02:03 luoyq6