Commit graph

69 commits

Author SHA1 Message Date
iperov
ba06a71fff SAE: changed default style power to 10.0 . Now style power is floating number in valid range 0.0 to 100.0 2019-01-14 21:24:33 +04:00
iperov
1f2b1481ef now you can train models on multiple GPU's on same workspace without cloning any folders.
Model files names will be prefixed with GPU index if GPU choosed explicitly on train/convert start.
if you leave GPU idx choice default, then best GPU idx will be choosed and model file names will not contain index prefix.
It gives you possibility to train same fake with various models or options on multiple GPUs.

H64 and H128: now you can choose 'Lighter autoencoder'. It is same as vram gb <= 4 before this update.

added archived_models.zip contains old experiments

RecycleGAN: archived

devicelib: if your system has no NVML installed (some old cards), then it will work with gpu_idx=0 as 'Generic GeForce GPU' with 2GB vram.

refactorings
2019-01-14 10:48:23 +04:00
iperov
e0605df175 upd readme 2019-01-13 17:47:05 +04:00
iperov
12383570e8 Fix for systems without NVSMI 2019-01-13 16:38:17 +04:00
iperov
81e837d920 upd readme 2019-01-12 15:29:34 +04:00
iperov
2473dabf37 fix loss history 2019-01-11 19:19:29 +04:00
iperov
bf5282f7ec if you have multi GPU in a system, you can choose what GPU to use on model start 2019-01-10 19:22:54 +04:00
iperov
48d0123f0b SAE: new options face_style_power, bg_style_power instead of style_power. Zero - means dont use style.
SAE: new option 'lighter_encoder'.
Now model settings can be overrided by pressing enter in 2 seconds while model loading (works on Windows).
Removed all MultiGPU models, because keras multi_gpu in fact doesn't work.
2019-01-10 18:28:12 +04:00
iperov
e8620919a7 update devicelib/nnlib to detect compute capability 2019-01-08 10:48:06 +04:00
iperov
7b70e7eec1 added new model U-net Face Morpher.
removed AVATAR - useless model was just for demo
removed MIAEF128 - use UFM insted
removed LIAEF128YAW - use model option sort by yaw on start for any model
All models now ask some options on start.
Session options (such as target epoch, batch_size, write_preview_history etc) can be overrided by special command arg.
Converter now always ask options and no more support to define options via command line.
fix bug when ConverterMasked always used not predicted mask.
SampleGenerator now always generate samples with replicated border, exclude mask samples.
refactorings
2019-01-02 17:26:12 +04:00
Plucky
a8694b73f0 DockerFile for Mac users to run DeepfaceLab with CPU Mode (#95)
* fix localization nullpointer exception

* fix devicelib error line:61,remove e

* support create docker from cpu dockerfile

* support preview or not when train(resolve cannot connect to X server)
2019-01-01 18:08:21 +04:00
iperov
474fff248f RecycleGAN fixes 2018-12-28 23:13:29 +04:00
iperov
1f86d7f1dd ResNet, changed PixelShuffler to Conv2DTranspose 2018-12-28 20:38:26 +04:00
iperov
f8824f9601 refactoring. Added RecycleGAN for testing. 2018-12-28 19:38:52 +04:00
iperov
e4010d3eba refactoring 2018-12-24 18:41:36 +04:00
iperov
8a223845fb refactoring 2018-12-24 13:45:40 +04:00
TooMuchFun
57fd6735a7 update to Keras 2.2.2 2018-11-14 07:22:27 -08:00
iperov
58d9b4ae09 fix avatar multigpu mode 2018-06-07 12:18:42 +04:00
iperov
6bd5a44264 initial 2018-06-04 17:12:43 +04:00