Hi, I trained three small models for beginners. He specially trained on a weak video card, as evidence that even at 1050ti you can get at least some acceptable result. However, these models can also work on more powerful graphics cards. I hope they help beginners save a couple of days on pre-training and selection of settings.
Also, these models can be launched not only at 1050ti, but also on any other high-level graphics card.
Used PC:
i5 8400 + 1050ti + 16gb ram
1) quick96_100k_iter:
Settings:
=============== Model Summary ================
== ==
== Model name: Quick96 ==
== ==
== Current iteration: 100062 ==
== ==
== ------------ Model Options ------------- ==
== ==
== batch_size: 4 ==
== ==
== -------------- Running On -------------- ==
== ==
== Device index: 0 ==
== Name: GeForce GTX 1050 Ti ==
== VRAM: 4.00GB ==
== ==
==============================================
Download link: https://mega.nz/#F!A5whCahR!QAqDzI4WKgELjY6EgTKKiw
_____________________
2) saehd-df-bs4_100k_iter:
Settings:
=============== Model Summary ===============
== ==
== Model name: SAEHD ==
== ==
== Current iteration: 100306 ==
== ==
==------------- Model Options -------------==
== ==
== random_flip: True ==
== resolution: 128 ==
== face_type: f ==
== learn_mask: False ==
== optimizer_mode: 2 ==
== archi: df ==
== ae_dims: 256 ==
== ed_ch_dims: 21 ==
== lr_dropout: False ==
== random_warp: False ==
== true_face_training: False ==
== face_style_power: 0.0 ==
== bg_style_power: 0.0 ==
== ct_mode: none ==
== clipgrad: True ==
== batch_size: 4 ==
== ==
==-------------- Running On ---------------==
== ==
== Device index: 0 ==
== Name: GeForce GTX 1050 Ti ==
== VRAM: 4.00GB ==
== ==
=============================================
Download link: https://mega.nz/#F!plwTnCqZ!TuoEomGb_FsN6xpxU_B4ug
_______________________
3) saehd-liae-bs4_160k_iter:
Settings:
=============== Model Summary ===============
== ==
== Model name: SAEHD ==
== ==
== Current iteration: 162841 ==
== ==
==------------- Model Options -------------==
== ==
== random_flip: True ==
== resolution: 128 ==
== face_type: f ==
== learn_mask: False ==
== optimizer_mode: 2 ==
== archi: liae ==
== ae_dims: 256 ==
== ed_ch_dims: 21 ==
== lr_dropout: False ==
== random_warp: False ==
== true_face_training: False ==
== face_style_power: 0.0 ==
== bg_style_power: 0.0 ==
== ct_mode: none ==
== clipgrad: True ==
== batch_size: 4 ==
== ==
==-------------- Running On ---------------==
== ==
== Device index: 0 ==
== Name: GeForce GTX 1050 Ti ==
== VRAM: 4.00GB ==
== ==
=============================================
Download link: https://mega.nz/#F!YlBVQAKa!UHV4U4ojXO_BkbjgTmzwiw
Also, these models can be launched not only at 1050ti, but also on any other high-level graphics card.
Used PC:
i5 8400 + 1050ti + 16gb ram
1) quick96_100k_iter:
Settings:
=============== Model Summary ================
== ==
== Model name: Quick96 ==
== ==
== Current iteration: 100062 ==
== ==
== ------------ Model Options ------------- ==
== ==
== batch_size: 4 ==
== ==
== -------------- Running On -------------- ==
== ==
== Device index: 0 ==
== Name: GeForce GTX 1050 Ti ==
== VRAM: 4.00GB ==
== ==
==============================================
Download link: https://mega.nz/#F!A5whCahR!QAqDzI4WKgELjY6EgTKKiw
_____________________
2) saehd-df-bs4_100k_iter:
Settings:
=============== Model Summary ===============
== ==
== Model name: SAEHD ==
== ==
== Current iteration: 100306 ==
== ==
==------------- Model Options -------------==
== ==
== random_flip: True ==
== resolution: 128 ==
== face_type: f ==
== learn_mask: False ==
== optimizer_mode: 2 ==
== archi: df ==
== ae_dims: 256 ==
== ed_ch_dims: 21 ==
== lr_dropout: False ==
== random_warp: False ==
== true_face_training: False ==
== face_style_power: 0.0 ==
== bg_style_power: 0.0 ==
== ct_mode: none ==
== clipgrad: True ==
== batch_size: 4 ==
== ==
==-------------- Running On ---------------==
== ==
== Device index: 0 ==
== Name: GeForce GTX 1050 Ti ==
== VRAM: 4.00GB ==
== ==
=============================================
Download link: https://mega.nz/#F!plwTnCqZ!TuoEomGb_FsN6xpxU_B4ug
_______________________
3) saehd-liae-bs4_160k_iter:
Settings:
=============== Model Summary ===============
== ==
== Model name: SAEHD ==
== ==
== Current iteration: 162841 ==
== ==
==------------- Model Options -------------==
== ==
== random_flip: True ==
== resolution: 128 ==
== face_type: f ==
== learn_mask: False ==
== optimizer_mode: 2 ==
== archi: liae ==
== ae_dims: 256 ==
== ed_ch_dims: 21 ==
== lr_dropout: False ==
== random_warp: False ==
== true_face_training: False ==
== face_style_power: 0.0 ==
== bg_style_power: 0.0 ==
== ct_mode: none ==
== clipgrad: True ==
== batch_size: 4 ==
== ==
==-------------- Running On ---------------==
== ==
== Device index: 0 ==
== Name: GeForce GTX 1050 Ti ==
== VRAM: 4.00GB ==
== ==
=============================================
Download link: https://mega.nz/#F!YlBVQAKa!UHV4U4ojXO_BkbjgTmzwiw