From 72fd7228c575c8ea90505a8fb35bd8346fc7affe Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 09:57:58 -0700 Subject: [PATCH] Cast to bool if using newer, int options --- models/Model_SAE/Model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index b15079a..ead0282 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -124,7 +124,7 @@ class SAEModel(ModelBase): default_apply_random_ct = False if is_first_run else self.options.get('apply_random_ct', False) self.options['apply_random_ct'] = io.input_bool( "Apply random color transfer to src faceset? (y/n, ?:help skip:%s) : " % ( - yn_str[default_apply_random_ct]), default_apply_random_ct, + yn_str[bool(default_apply_random_ct)]), bool(default_apply_random_ct), help_message="Increase variativity of src samples by apply LCT color transfer from random dst samples. It is like 'face_style' learning, but more precise color transfer and without risk of model collapse, also it does not require additional GPU resources, but the training time may be longer, due to the src faceset is becoming more diverse.") if nnlib.device.backend != 'plaidML': # todo https://github.com/plaidml/plaidml/issues/301