1 files changed, 1 insertions, 1 deletions

diff --git a/train_ti.py b/train_ti.py
index 7f93960..1dbd637 100644
--- a/train_ti.py
+++ b/train_ti.py
@@ -923,7 +923,7 @@ def main():
             progressive_buckets=args.progressive_buckets,
             bucket_step_size=args.bucket_step_size,
             bucket_max_pixels=args.bucket_max_pixels,
-            dropout=args.tag_dropout,
+            tag_dropout=args.tag_dropout,
             shuffle=not args.no_tag_shuffle,
             template_key=data_template,
             placeholder_tokens=args.placeholder_tokens,