github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-19 13:09:56 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'amp_updates' into amp_test_config

Browse source 
# Conflicts:
#	models/Model_AMP/Model.py
This commit is contained in:
Jan 2021-12-05 18:21:05 +01:00
commit ba8ccf732a
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/Model_AMP/Model.py
View file

@ -62,13 +62,14 @@ class AMPModel(ModelBase):
if self.is_first_run() or ask_override:
if (self.read_from_conf and not self.config_file_exists) or not self.read_from_conf:
self.ask_autobackup_hour()
self.ask_maximum_n_backups()
self.ask_write_preview_history()
self.ask_target_iter()
self.ask_retraining_samples()
self.ask_random_src_flip()
self.ask_random_dst_flip()
self.ask_batch_size(8)
# self.options['use_fp16'] = io.input_bool ("Use fp16", default_usefp16, help_message='Increases training/inference speed, reduces model size. Model may crash. Enable it after 1-5k iters.')
self.options['use_fp16'] = io.input_bool ("Use fp16", default_usefp16, help_message='Increases training/inference speed, reduces model size. Model may crash. Enable it after 1-5k iters.')
if self.is_first_run():