convert_2d_to_3d

BloodAxe · Feb 6, 2025 · 28a3791 · 28a3791
1 parent e67ffa5
commit 28a3791
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch_toolbelt/utils/torch_utils.py b/pytorch_toolbelt/utils/torch_utils.py
@@ -47,6 +47,7 @@
     "get_collate_for_dataset",
     "get_non_wrapped_model",
     "container_to_tensor",
+    "convert_2d_to_3d",
 ]