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

amp force preview sample and prop change

Browse source
This commit is contained in:
seranus 2021-12-09 18:17:46 +01:00
commit 790c888c5e
4 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
models/Model_AMP/Model.py
View file

@ -933,8 +933,10 @@ 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] )]
#n_samples = min(4, self.get_batch_size(), 800 // self.resolution ) 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 = []

5
models/Model_AMP/config_schema.json
View file

@ -179,6 +179,11 @@
"random_dst_flip": { "random_dst_flip": {
"type": "boolean" "type": "boolean"
}, },
"force_preview_samples_num": {
"type": "integer",
"minimum": 1,
"maximum": 64
},
"batch_size": { "batch_size": {
"type": "integer", "type": "integer",
"minimum": 1 "minimum": 1

4
models/Model_SAEHD/Model.py
View file

@ -979,10 +979,10 @@ class SAEHDModel(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['preview_samples'] is None: if self.options['force_preview_samples_num'] is None:
n_samples = min(4, self.get_batch_size(), 800 // self.resolution ) n_samples = min(4, self.get_batch_size(), 800 // self.resolution )
else: else:
n_samples = min(self.get_batch_size(), self.options['preview_samples']) n_samples = min(self.get_batch_size(), self.options['force_preview_samples_num'])
if filenames is not None and len(filenames) > 0: if filenames is not None and len(filenames) > 0:
for i in range(n_samples): for i in range(n_samples):

2
models/Model_SAEHD/config_schema.json
View file

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