这几天又更新了,好事儿是又有新功能可以玩了,“坏事儿”是感觉又要升级设备了。其实,总的来说这次更新,对低配高配玩家都有好消息。
1. 对于高配玩家,可以跑更高像素的模型。
2. 对于低配玩家,那些跑不起的像素也能跑起来了。
下面就来说说重点更新内容:
1. 模型训练参数 resolution 的最大值从512调到了640。
这意味着对于高配玩家来说可以跑更大的模型,大到640 x640 。这是一个什么样的概念呢? 比如一个720P的视频分辨率是1280x720,现在模型的分辨率就可以达到640,也就是这个级别的视频基本上特效镜头都能涵盖到,即便是1080P,只要配合超分也不会出现模糊的问题(模型训练好的前提下)。
2. 模型结构参数调整,移除HD,添加添加 –D
这是什么意思呢?HD效果不好,所以移除了,而–D 这个就厉害了。根据更新提示,这个结构可以在同样的配置下跑两倍的像素,比如原先跑224,现在可以跑448 。或者同样的配置,跑同样的参数,可以快一倍。也就是整体性能提高了一倍。这一点很容易验证,至于效果上有没有下降,这个还是需要时间去验证,毕竟这也是一个试验性的参数。
具体的使用方法就是在训练新模型的时候模型结构(AE architecture)这一栏设置成df–d 或者liae–d。
这是用滴滴云P4 8g 跑结构为df–d 像素为256模型。 如果使用DF结构的话8G显卡跑256,bs8都起不来,而DF–D结构可以轻松开到bs16。
除了上面的两大更新之外,还有优化了gan_power选项,添加了Uniform yaw distribution 选项,优化了quick96模型,优化了xseg等。详细的更新内容见:
另外免费的colab脚本也已经添加了601和619的版本。