MrDeepFakes Forums

Some content may not be available to Guests. Consider registering an account to enjoy unrestricted access to guides, support and tools

  • We are looking for community members who are intested in helping out. See our HELP WANTED post.

oom on new SAE, when turning off learning mask

Mr.BaBe

DF Enthusiast
Verified Video Creator
hello

I am using 256 resolution pre made model for SAE, Downloaded from the forums. 
I have 8 gig vram,(RTX2070), when I train with batch size 5, with learning on, warp and true face on, model trains without any problem, but when I turn off learning, keep other two on, or warp off true face on (at 4) , I end up oom messages. I drop down true face to 3, still oom. 
how come it model trains with learning ,warp and true face on , but not works with learning, warp off, only true face on, with all same settings

thanks in advance
 

Groggy4

NotSure
Verified Video Creator
Kind of amazed you manage batch 5 at all with 8gb card. What's the dims on that model?
 

Mr.BaBe

DF Enthusiast
Verified Video Creator
Groggy4 said:
Kind of amazed you manage batch 5 at all with 8gb card. What's the dims on that model?

============= Model Summary ==============
==                                      ==
==         Model name: SAE              ==
==                                      ==
==  Current iteration: 263032           ==
==                                      ==
==----------- Model Options ------------==
==                                      ==
==        sort_by_yaw: False            ==
==        random_flip: False            ==
==         resolution: 256              ==
==          face_type: f                ==
==         learn_mask: True             ==
==     optimizer_mode: 3                ==
==              archi: liae             ==
==            ae_dims: 256              ==
==          e_ch_dims: 21               ==
==          d_ch_dims: 21               ==
==         ca_weights: True             ==
==         pixel_loss: False            ==
==        random_warp: True             ==
==   face_style_power: 0.0              ==
==     bg_style_power: 0.0              ==
==            ct_mode: none             ==
==           clipgrad: False            ==
==         batch_size: 5                ==
== true_face_training: 4                ==
==            network: rmsprop          ==
==                                      ==
==------------- Running On -------------==
==                                      ==
==       Device index: 0                ==
==               Name: GeForce RTX 2070 ==
==               VRAM: 8.00GB           ==
==                                      ==
==========================================

with this setting no oom, trains ok, but when I turn off learning mask and wrap face, train does not work, unless i reduce batch size from 5 to 4. its weird, since learnin g mask consumes a lot of ram and turning it off should give me extra ram,
 
Top