增加一些设置。

train/train_embedding.py

@@ -660,7 +660,7 @@ def main(task_key: str = "dish"):
         batch_size=BATCH_SIZE, 
         shuffle=False, 
         num_workers=0,
-        drop_last=True
+        drop_last=False
     )
     
     logger.info(f"训练集大小: {len(train_dataset)}")
@@ -811,4 +811,4 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("task", choices=list(TASKS.keys()), nargs="?", default="dish")
     args = parser.parse_args()
-    main(args.task)
+    main("whole_ingredient")