In HoloEncoder_training.m show some error

I use this program "HoloEncoder_training.m" want to training,

but show the error ：
"Error using dlfeval (line 43) Layer 'I': Invalid input data. Error using 'predict' in Layer fresnelLayer. The function threw an error and could not be executed."

Have anyone same problem in this training part?

I used GPU RTX-3090 & Matlab 2020b version