From a07f04599aa9dfa0f54d5ce1f00a616447f0eec2 Mon Sep 17 00:00:00 2001
From: Sean Lee <xmlee97@gmail.com>
Date: Mon, 30 Sep 2024 09:18:28 +0800
Subject: [PATCH] remove_unused_columns=None

---
 angle_emb/angle.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/angle_emb/angle.py b/angle_emb/angle.py
index c5a60e6..f45ab59 100644
--- a/angle_emb/angle.py
+++ b/angle_emb/angle.py
@@ -1545,6 +1545,7 @@ def fit(self,
                 save_total_limit=save_total_limit,
                 load_best_model_at_end=False,
                 ddp_find_unused_parameters=False if self.gpu_count > 1 else None,
+                remove_unused_columns=False,
                 **argument_kwargs,
             ),
             callbacks=callbacks,