Fail to run pad_experiment

[trid2912]

Hi, thanks for your detail instructions to run the code.
When trying to run the pad_experiment based on your instruction, I met this bug

Here is the script that I try to run
python pad_experiments.py --train --model convlstm --prefix_coco "20231130035643" --num_epochs 10 --batch_size 32 --bands B02 B03 B04 B08 --saved_medians --img_size 61 61 --requires_norm --num_workers 16 --fixed_window
I run this code with 1 GPU with 24GB RAM
Thanks for your help!