From 2469501c3951a9ed86c820cddf7b32144a4a1c8d Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 19 Jan 2023 09:04:39 +0100 Subject: Move Accelerator preparation into strategy --- train_ti.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 7aa4960..451b61b 100644 --- a/train_ti.py +++ b/train_ti.py @@ -159,7 +159,7 @@ def parse_args(): parser.add_argument( "--tag_dropout", type=float, - default=0.1, + default=0, help="Tag dropout probability.", ) parser.add_argument( @@ -407,7 +407,7 @@ def parse_args(): ) parser.add_argument( "--emb_decay", - default=1e-2, + default=10, type=float, help="Embedding decay factor." ) @@ -597,7 +597,7 @@ def main(): def run(i: int, placeholder_tokens, initializer_tokens, num_vectors, data_template): if len(placeholder_tokens) == 1: - sample_output_dir = output_dir.joinpath(f"samples_{placeholder_token[0]}") + sample_output_dir = output_dir.joinpath(f"samples_{placeholder_tokens[0]}") else: sample_output_dir = output_dir.joinpath("samples") -- cgit v1.2.3-54-g00ecf