准备利用K3硬件平台训练一个AI模型，需要这个包，另外,如果有如下预训练的AI模型也请提供给我：
一个 U-Net 风格的图像到图像翻译模型：输入低分辨率颜色+深度+运动矢量，输出高分辨率抗锯齿画面。
L1 loss + perceptual loss (VGG)
这是我们目前的unet模型：
你可以基于该模板修改适配你的自定义模型结构和输入输出。
这两个文档可以参考：
https://www.spacemit.com/community/document/info?nodepath=competition/2026嵌入式大赛/比赛产品包/05_Model_Zoo.md&lang=zh
https://www.spacemit.com/community/document/info?nodepath=software/SDK/ros/k1/04_Model_deployment/4.1_Demo_Zoo_Overview.md&lang=zh
有没有pytorch预编译包？我想自己用k3训练一个
官方推荐的模板是语义分割模型，我需要的是图像翻译模型，两者不是一回事，无法通过修改适配。
查看了一下官方的model zoo,没有合适的prototype,我自己试着构建一个模型训练器吧，不需要pytorch了，谢啦