How to solve error in model_builder.py?

fanq15 / FSOD-code

MIT License

244 stars 61 forks source link

How to solve error in model_builder.py? #22

Open WangSong960913 opened 4 years ago

WangSong960913 commented 4 years ago

Expected 4-dimensional input for 4-dimensional weight [64, 3, 7, 7], but got input of size [8, 10, 3, 320, 320] instead original_blob_conv = self.Conv_Body(im_data) support_blob_conv = self.Conv_Body(support_data.squeeze(0))# im_data is a data with 4-dimensional (8,3,600,800 ) support_data is a data with 5-dimensional(8,10,3,320,320 )

fanq15 commented 4 years ago

It seems that you set more than 1 query image in a GPU. You need to make sure there is only one query image in one GPU.