@liwi 5 April 2021 more from that month

關於 zi2zi / pix2pix，把 tensorflow-gpu 版本移除改回 cpu 版本了。 加上一些筆記，給想接著玩的人參考。我是 ML/DL 菜鳥，所以參考性請自行拿捏。歡迎大神指教。 Code 在這裡 https://github.com/kaonashi-tyc/zi2zi 基本上就是整個打包，弄成他要的環境就可以跑起來了。 記憶體的部分用預設的 batch_size=16 的話大概 3 GB 可以跑起來，一個 batch 大概 150 ~ 200 秒，我照預設的餵了 1000 個 sample，一半 train 一半 validate，等於 500 對，所以一個 epoch 有 32 個 batch 要跑。 算起來差不多一個 sample 10 秒左右。500 對就是 5000 秒，也就是差不多 1.5 小時一個 epoch。目前設 10 個 epoch 給他跑。跑完來 infer 看看字長怎樣。 原文用 1080 GTX 跑了 29000 個 sample，超過 30 個 epoch，花了兩天，換算起來 0.2 秒一個 sample，是我 cpu 跑速的 ~50 倍（很合理）。 記得，要認真跑的話可以先看看怎麼載原作已經調教過的 model，應該可以省卻很多訓練量。我之後可能也會嘗試用 cpu 結合原作給的 model 來微調成想要的字形跟玩 interpolation。 3 GB 就可以跑的話不太懂為什麼我 batch 改到 4，2 GB（實際可用約 1.5 GB 的樣子） 的 gpu 還是跑不起來，可能有一些 overhead 跟 batch 大小無關。看了一下一般似乎建議用 8 或 16 的 batch size 開始跑，所以用 4 跑可能也只是浪費時間。 #PttDigiCurrency