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?

Commencez par exécuter auto3dseg_hello_world.ipynb avec les chemins et paramètres indiqués, en comparant les jeux de données de 20 et 300 images dans un environnement avec un GPU de 48G. Examinez l’étape get_data et l’endroit où l’entraînement génère « Pin memory thread exited unexpectedly. ». C’est terminé lorsque le notebook peut entraîner le modèle sur le jeu de données le plus volumineux sans cette erreur, tout en préservant le comportement avec le jeu de données plus petit.