FSOD-code icon indicating copy to clipboard operation
FSOD-code copied to clipboard

How to solve error in model_builder.py?

Open WangSong960913 opened this issue 5 years ago • 1 comments

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 )

WangSong960913 avatar Jun 13 '20 13:06 WangSong960913

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.

fanq15 avatar Jun 15 '20 03:06 fanq15