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

amp - preview sample fix

Browse source
This commit is contained in:
seranus 2021-12-09 18:27:47 +01:00
commit 1b521674c5
2 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
models/Model_AMP/Model.py
View file

@ -69,7 +69,7 @@ class AMPModel(ModelBase):
self.ask_session_name() self.ask_session_name()
self.ask_maximum_n_backups() self.ask_maximum_n_backups()
self.ask_write_preview_history() self.ask_write_preview_history()
self.options['preview_samples'] = np.clip ( io.input_int ("Number of samples to preview", default_cpu_cap, add_info="1 - 16", help_message="Typical fine value is 4"), 1, 16 ) self.options['preview_samples'] = np.clip ( io.input_int ("Number of samples to preview", default_preview_samples, add_info="1 - 16", help_message="Typical fine value is 4"), 1, 16 )
self.ask_target_iter() self.ask_target_iter()
self.ask_retraining_samples() self.ask_retraining_samples()
self.ask_random_src_flip() self.ask_random_src_flip()
@ -933,10 +933,7 @@ class AMPModel(ModelBase):
target_srcm, target_dstm = [ nn.to_data_format(x,"NHWC", self.model_data_format) for x in ([target_srcm, target_dstm] )] target_srcm, target_dstm = [ nn.to_data_format(x,"NHWC", self.model_data_format) for x in ([target_srcm, target_dstm] )]
if self.options['force_preview_samples_num'] is None: n_samples = min(self.get_batch_size(), self.options['preview_samples'])
n_samples = min(4, self.get_batch_size(), 800 // self.resolution )
else:
n_samples = min(self.get_batch_size(), self.options['force_preview_samples_num'])
result = [] result = []

2
models/Model_AMP/config_schema.json
View file

@ -179,7 +179,7 @@
"random_dst_flip": { "random_dst_flip": {
"type": "boolean" "type": "boolean"
}, },
"force_preview_samples_num": { "preview_samples": {
"type": "integer", "type": "integer",
"minimum": 1, "minimum": 1,
"maximum": 64 "maximum": 64