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?

まず、報告されているパスとパラメータで auto3dseg_hello_world.ipynb を実行し、48G GPU 環境で 20 画像のデータセットと 300 画像のデータセットを比較します。get_data ステップと、トレーニングで “Pin memory thread exited unexpectedly.” が発生する箇所を調査します。大きいデータセットでこのエラーなしにノートブックのトレーニングを実行でき、小さいデータセットでの動作が維持されれば完了です。