HF-Avatar icon indicating copy to clipboard operation
HF-Avatar copied to clipboard

关于octopus_weights.hdf5权重的问题

Open muximuxi opened this issue 1 year ago • 1 comments

您好! 根据提示将octopus_weights.hdf5下载并放到weights文件夹下,使用sample_data, 并执行第一步python _infer_single.py后,报错: ValueError: Layer #5 (named "conv2d_1"), weight <tf.Variable 'conv2d_1/kernel:0' shape=(3, 3, 8, 16) dtype=float32> has shape (3, 3, 8, 16), but the saved weight has shape (8, 3, 3, 3). 请问是哪里出问题了吗?

muximuxi avatar Sep 19 '23 09:09 muximuxi

可以把octopus文件中,286行的by_name改成False试一试,

alpttex19 avatar Dec 08 '23 02:12 alpttex19