In HoloEncoder_training.m show some error

THUHoloLab / Holo-encoder

High-speed computer-generated holography (CGH) using autoencoder-based deep neural network

49 stars 10 forks source link

In HoloEncoder_training.m show some error #2

Open HunagYuJie opened 2 years ago

HunagYuJie commented 2 years ago

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