lyuwenyu
commented
6 months ago - 提供的
paddle转的参数只包含ema的结果，不完全对应torch训练时候的所有的state - eval的时候使用的是solver ema的module, 所以参数应该加载到solver的ema里 不应该是模型本身
Open mgh233 opened 6 months ago
lyuwenyu
commented
6 months ago paddle转的参数只包含ema的结果，不完全对应torch训练时候的所有的state
首先是模型在resume过程中出错，具体原因是模型的字典没有对应上。
如上图所示，下载得到的模型的权重对应在module中而不是model中，并且model应该在第一层级上而不是在ema中。 以上问题我通过修改key值能够解决。 其次是将修改后的模型进行test，得到的结果为0，如下图所示。
测试从pytorch代码中训练出来的模型则具有正常结果，如下图所示。