mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 04:52:13 -07:00
refactoring
This commit is contained in:
parent
121c0cfc0f
commit
eda6433936
5 changed files with 9 additions and 13 deletions
|
@ -30,7 +30,7 @@ class SampleGeneratorFace(SampleGeneratorBase):
|
|||
raise_on_no_data=True,
|
||||
**kwargs):
|
||||
|
||||
super().__init__(samples_path, debug, batch_size)
|
||||
super().__init__(debug, batch_size)
|
||||
self.sample_process_options = sample_process_options
|
||||
self.output_sample_types = output_sample_types
|
||||
self.add_sample_idx = add_sample_idx
|
||||
|
@ -40,7 +40,7 @@ class SampleGeneratorFace(SampleGeneratorBase):
|
|||
else:
|
||||
self.generators_count = max(1, generators_count)
|
||||
|
||||
samples = SampleLoader.load (SampleType.FACE, self.samples_path)
|
||||
samples = SampleLoader.load (SampleType.FACE, samples_path)
|
||||
self.samples_len = len(samples)
|
||||
|
||||
self.initialized = False
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue