search

Hi-FT / ERD

Overcoming Catastrophic Forgetting in Incremental Object Detection via Elastic Response Distillation
Apache License 2.0
92 stars 10 forks source link

How to do multi-step training #19

Open miaomiaojun122 opened 4 months ago

miaomiaojun122 commented 4 months ago

When I do 40+20+20 multi-step training, I always report mistakes, tell me unexpected key in source state_dict: ori_model.backbone.conv1.weight, ori_model.backbone.bn1.weight, ori_model.backbone.bn1.bias, ori_model.backbone.bn1.running_mean, ori_model.backbone.bn1.running_var, ori_model.backbone.bn1.num_batches_tracked, ori_model.backbone.layer1.0.conv1.weight, ori_model.backbone.layer1.0.bn1.weight, ori_model.backbone.layer1.0.bn1.bias, ori_model.backbone.layer1.0.bn1.running_mean, ori_model.backbone.layer1.0.bn1.running_var, ori_model.backbone.layer1.0.bn1.num_batches_tracked, ori_model.backbone.layer1.0.conv2.weight, ori_model.backbone.layer1.0.bn2.weight, ori_model.backbone.layer1.0.bn2.bias, ori_model.backbone.layer1.0.bn2.running_mean, ori_model.backbone.layer1.0.bn2.running_var, ori_model.backbone.layer1.0.bn2.num_batches_tracked, ori_model.backbone.layer1.0.conv3.weight, ori_model.backbone.layer1.0.bn3.weight, ori_model.backbone.layer1.0.bn3.bias, ori_model.backbone.layer1.0.bn3.running_mean, ori_model.backbone.layer1.0.bn3.running_var, ori_model.backbone.layer1.0.bn3.num_batches_tracked, ori_model.backbone.layer1.0.downsample.0.weight, ...... How to do multi-step training thank you