From e3669927b47b5367a3348d30c4b318da84af661d Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 2 Apr 2023 11:14:03 +0200 Subject: Update dataset format: Separate prompt and keywords --- train_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 5482326..651dfbe 100644 --- a/train_ti.py +++ b/train_ti.py @@ -186,7 +186,7 @@ def parse_args(): parser.add_argument( "--tag_dropout", type=float, - default=0, + default=0.1, help="Tag dropout probability.", ) parser.add_argument( -- cgit v1.2.3-54-g00ecf