search

yang-song / score_inverse_problems

Official repo for "Solving Inverse Problems in Medical Imaging with Score-Based Generative Models"
214 stars 26 forks source link

flax.errors.ScopeParamShapeError in ResnetBlockBigGANpp #9

Open PengbinYu opened 1 year ago

PengbinYu commented 1 year ago

Hello! When I was executing python main.py --config score_inverse_problems-main/configs/ve/brats_ncsnpp_baseline.py --eval_folder eval --mode eval --workdir workdirs, I encountered the following error：

image

Can you help me to find out the reason? Since my device is only equipped with GPU, my Jax version is 0.3.25, and the Jax-lib version is 0.3.25+cuda11.cudnn82, flax == 0.4.0 . Thank you so much！

sjhjkkjhjknn commented 1 year ago

I meet the same question with you, do you solve it now?

DeepXuan commented 1 year ago

Hi, I encountered the same issue, have you solved it?