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

fix missing use_fp16 var

Browse source
This commit is contained in:
JanFschr 2021-11-23 11:01:18 +01:00 • committed by GitHub
commit 262bfda8f8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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

@ -51,6 +51,7 @@ class AMPModel(ModelBase):
default_ct_mode = self.options['ct_mode'] = self.load_or_def_option('ct_mode', 'none')
default_random_color = self.options['random_color'] = self.load_or_def_option('random_color', False)
default_clipgrad = self.options['clipgrad'] = self.load_or_def_option('clipgrad', False)
default_use_fp16 = self.options['use_fp16'] = self.load_or_def_option('use_fp16', False)
ask_override = self.ask_override()
if self.is_first_run() or ask_override:
@ -167,7 +168,7 @@ class AMPModel(ModelBase):
adabelief = self.options['adabelief']
# use_fp16 = self.options['use_fp16']
use_fp16 = self.options['use_fp16']
if self.is_exporting:
use_fp16 = io.input_bool ("Export quantized?", False, help_message='Makes the exported model faster. If you have problems, disable this option.')