AFI-ForwardDeduplicate icon indicating copy to clipboard operation
AFI-ForwardDeduplicate copied to clipboard

how to use latest RIFE in AFI?

Open 2994dd42d75b0 opened this issue 6 months ago • 3 comments

I copied all files from RIFE 4.18 and also change paths in .py files Still getting this error

Traceback (most recent call last): File "/AFI-ForwardDeduplicate/interpolate_video_forward.py", line 93, in model.load_state_dict(convert(torch.load('weights/flownet.pkl'))) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 2189, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for IFNet: Unexpected key(s) in state_dict: "teacher.conv0.0.0.weight", "teacher.conv0.0.0.bias", "teacher.conv0.1.0.weight", "teacher.conv0.1.0.bias", "teacher.convblock.0.beta", "teacher.convblock.0.conv.weight", "teacher.convblock.0.conv.bias", "teacher.convblock.1.beta", "teacher.convblock.1.conv.weight", "teacher.convblock.1.conv.bias", "teacher.convblock.2.beta", "teacher.convblock.2.conv.weight", "teacher.convblock.2.conv.bias", "teacher.convblock.3.beta", "teacher.convblock.3.conv.weight", "teacher.convblock.3.conv.bias", "teacher.convblock.4.beta", "teacher.convblock.4.conv.weight", "teacher.convblock.4.conv.bias", "teacher.convblock.5.beta", "teacher.convblock.5.conv.weight", "teacher.convblock.5.conv.bias", "teacher.convblock.6.beta", "teacher.convblock.6.conv.weight", "teacher.convblock.6.conv.bias", "teacher.convblock.7.beta", "teacher.convblock.7.conv.weight", "teacher.convblock.7.conv.bias", "teacher.lastconv.0.weight", "teacher.lastconv.0.bias".

I renamed rife48.pkl to flownet.pkl Please tell how to fix this

2994dd42d75b0 avatar Aug 03 '24 18:08 2994dd42d75b0