Added FIXME

load_data.py

@@ -44,6 +44,7 @@ def load_data_finetune(args, tokenizer):
     dataset = load_dataset(*arg_map[args.dataset])
 
     def tokenize_function_finetune(batch):
+        # FIXME This fails for GLUE MNLI
         return tokenizer(batch["sentence"], truncation=True, padding="max_length", max_length=args.max_length)
     
     dataset = dataset.map(tokenize_function_finetune)