使用我自己的自定义数据集进行视频帧插值

问：

我想用训练我自己的自定义模型来制作一个 PyTorch 程序。使用 RIFE 预训练模型（链接：文本）有一个很好的程序。但我想将模型替换为我自己的模型，或者微调预训练的数据集。我不知道如何组织自定义模型并从 RIFE 预训练模型中替换它。示例程序太复杂了，我很难理解。能理解该程序的人请给我一些建议！:(

我希望通过添加自己的数据集和操作（微调）预训练模型从 PyTorch 进行视频帧插值。我试图理解示例代码，但太难理解了！