can't run on large datasets

When I correctly follow the steps in the auto3dseg_hello_world.ipynb notebook, set the corresponding paths and parameters, and run it in an environment with 48G of GPU memory, I encounter the error RuntimeError: Pin memory thread exited unexpectedly while attempting to train on a dataset with 300 .nii.gz images. In contrast, when using a dataset with 20 images, the training proceeds smoothly under the exact same conditions. During the training process with the 300-image dataset, I monitored the GPU memory usage and found it to be less than 70%. However, the error keeps occurring inexplicably. Could there be an issue with the get_data step?

Beginnen Sie damit, auto3dseg_hello_world.ipynb mit den gemeldeten Pfaden und Parametern auszuführen und die Datensätze mit 20 und 300 Bildern in einer 48G-GPU-Umgebung zu vergleichen. Untersuchen Sie den get_data-Schritt und die Stelle, an der das Training „Pin memory thread exited unexpectedly.“ auslöst. Als erledigt gilt die Aufgabe, wenn das Notebook auf dem größeren Datensatz ohne diesen Fehler trainieren kann und das Verhalten beim kleineren Datensatz erhalten bleibt.