remove_unused_columns=None

SeanLee97 · Sep 30, 2024 · a07f045 · a07f045
1 parent 228ae65
commit a07f045
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,