1 files changed, 3 insertions, 3 deletions

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")