验证成功导入模型

    for target_key, target_value in target_state.items():

k = target_key
print(target_key)
输出模型的所有关键字,判断当时处理的哪个模型并往下边if走输出处理结果。发现torch.loadstatedict并没有报错,并没有发生缺少关键字或维度不匹配问题。
posted @ 2023-03-11 17:46  祥瑞哈哈哈  阅读(13)  评论(0)    收藏  举报