👕 Remove warning for format_batch_fn double save

ENSTA-U2IS-AI · Aug 25, 2023 · f98d2c8 · f98d2c8
1 parent a9ad38e
commit f98d2c8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_uncertainty/routines/classification.py b/torch_uncertainty/routines/classification.py
@@ -70,6 +70,7 @@ def __init__(
                 "model",
                 "loss",
                 "optimization_procedure",
+                "format_batch_fn",
             ]
         )