|
发表于 2021-8-13 20:10:30
|
显示全部楼层
How to create RTM model from scratch.
Using SAEHD model.
res:224, WF, ae_dims:256, e_dims:64, d_dims:64, d_mask_dims 22, eyes_mouth_prio:Y, batch 8. Others are default.
1) enable pretrain mode. Train to 1kk
2) disable pretrain mode, or use already trained RTM model and delete INTER_AB.npy file . Train to 1kk
3) lrd:N uniform_yaw:True, color_transfer:lct, train +500..800k
4) lrd:Y train +500..800k iters
5) disable random warp +500..800k iters
6) enable gan 0.1 gan_dims:32, train +100..300k iters
如何从头开始创建 RTM 模型。
使用 SAEHD 模型。
res:224, WF, ae_dims:256, e_dims:64, d_dims:64, d_mask_dims 22, eyes_mouth_prio:Y, batch 8. 其他为默认值。
1)启用预训练模式。训练到 1kk
2) 禁用预训练模式,或使用已经训练好的 RTM 模型并删除 INTER_AB.npy 文件。训练到 1kk
3) lrd :N uniform_yaw:True, color_transfer:lct, train +500..800k
4) lrd :Y train +500..800k iters
5) 禁用随机扭曲 +500..800k iters
6) 启用 gan 0.1 gan_dims:32, 训练 +100..300k iters |
|