Is the weight file wrong?

[NiFangBaAGe]

Hi,

I am very interested in SPAN. However, I find something wrong when I try to run it.

Traceback (most recent call last):
File "evaluator.py", line 39, in
model.load_weights(weight_file)
File "/home/ipprlab/anaconda3/envs/py3/lib/python3.6/site-packages/keras/engine/network.py", line 1166, in load_weights
f, self.layers, reshape=reshape)
File "/home/ipprlab/anaconda3/envs/py3/lib/python3.6/site-packages/keras/engine/saving.py", line 1030, in load_weights_from_hdf5_group
str(len(filtered_layers)) + ' layers.')
ValueError: You are trying to load a weight file containing 1 layers into a model with 12 layers.

Is the weight file wrong?

Thanks.

[siweic0818]

Did you use model = keras.utils.multi_gpu_model( model, 2)? The saved weights are created by multiple gpus so if you are using single gpu then the weights won't load.

[brightendavid]

is there any weight saved in single gpu?to get multi gpu is to hard for me.

[michiganleon]

Dear David, You can try to load the model in multi-gpu model and then transform the model into single GPU. For tensorflow you can check this code: https://stackoverflow.com/questions/41342098/keras-load-checkpoint-weights-hdf5-generated-by-multiple-gpus <https://stackoverflow.com/questions/41342098/keras-load-checkpoint-weights-hdf5-generated-by-multiple-gpus> This is keras but I remember normal TF model can be loaded similarly. (Using model.layers[-2]) Sincerely, Xuefeng

…

2021年9月1日 下午11:35，brightendavid ***@***.***> 写道： is there any weight saved in single gpu?to get multi gpu is to hard for me. — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#4 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ADI7BMHRDCPSW6VFUMQTEBLT74LMFANCNFSM4YXYF4TA>. Triage notifications on the go with GitHub Mobile for iOS <https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675> or Android <https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub>.

[85zhanghao]

I have the same problem, have you solved it ?“ValueError: You are trying to load a weight file containing 1 layers into a model with 12 layers.”