Error loading weights of NCPDNet-3D
#1
by
yitongyang997
- opened
Dear author,
I got the following error when loading the model weights: ValueError: Cannot assign value to variable ' ncpd_net/image_net_0/conv3d/kernel:0': Shape mismatch.The variable shape (3, 3, 3, 12, 16), and the assigned value shape (3, 3, 3, 6, 16) are incompatible. I am running google colab with tensorflow version 2.8.
Hi @yitongyang997 ,
Thanks for your interest in this model and raising this issue.
Indeed, the code to load the weights was bugged: one needs to specify the number of primal variables n_primal=2 (as mentioned in the paper https://inria.hal.science/hal-03188997v3/document).
I will correct it in the model card.
Cheers,
Zaccharie
zaccharieramzi
changed discussion status to
closed