這幾天又更新了,好事兒是又有新功能可以玩了,“壞事兒”是感覺又要升級設備了。其實,總的來說這次更新,對低配高配玩家都有好消息。
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的版本。