Personal pytorch devkit.
A fast ssim & ms-ssim implement code with pytorch jit.
I added a custom backward function that can really save your VRAM. At the same time can be compatible with other normal model. A network similar to i-revnet saves a lot of video memory.
最近一年贡献:0 次
最长连续贡献:0 日
最近连续贡献:0 日
贡献度的统计数据包括代码提交、创建任务 / Pull Request、合并 Pull Request,其中代码提交的次数需本地配置的 git 邮箱是 Gitee 帐号已确认绑定的才会被统计。