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

Export AMP/SAEHD: added "Export quantized" option

Browse source
This commit is contained in:
iperov 2021-07-30 17:13:46 +04:00
parent f5cc54177f
commit 83b1412da7
2 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/Model_AMP/Model.py
View file

@ -125,7 +125,10 @@ class AMPModel(ModelBase):
if ct_mode == 'none':
ct_mode = None
use_fp16 = self.is_exporting
use_fp16 = False
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.')
conv_dtype = tf.float16 if use_fp16 else tf.float32
class Downscale(nn.ModelBase):