From c55920681ccd15c183f168619d9b332cd837163c Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 3 Jan 2023 13:32:37 +0100 Subject: Don't use vector_dropout by default --- train_dreambooth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index f26b7f5..df8b54c 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -110,7 +110,7 @@ def parse_args(): parser.add_argument( "--vector_dropout", type=int, - default=0.1, + default=0, help="Vector dropout probability.", ) parser.add_argument( -- cgit v1.2.3-54-g00ecf