From 532552650b3fa0c359469a61e363a16bc83eb094 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Wed, 8 Mar 2023 19:33:53 -0500 Subject: [PATCH] Fix issue 335 --- library/common_gui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/common_gui.py b/library/common_gui.py index c238a11..c42c61d 100644 --- a/library/common_gui.py +++ b/library/common_gui.py @@ -790,7 +790,7 @@ def run_cmd_advanced_training(**kwargs): f' --keep_tokens="{kwargs.get("keep_tokens", "")}"' if int(kwargs.get('keep_tokens', 0)) > 0 else '', - f' --caption_dropout_every_n_epochs="{kwargs.get("caption_dropout_every_n_epochs", "")}"' + f' --caption_dropout_every_n_epochs="{int(kwargs.get("caption_dropout_every_n_epochs", 0))}"' if int(kwargs.get('caption_dropout_every_n_epochs', 0)) > 0 else '', f' --caption_dropout_rate="{kwargs.get("caption_dropout_rate", "")}"'