Incompatible tensor size when a running a batch size of 2

yrcong / RelTR

RelTR: Relation Transformer for Scene Graph Generation: https://arxiv.org/abs/2201.11460v2

248 stars 49 forks source link

Incompatible tensor size when a running a batch size of 2 #56

Closed ikram-md closed 4 months ago

ikram-md commented 4 months ago

I successfully managed to solve the environment issue, yet now when i tried to launch the training, i am obtaining the following error : And when i use batch size of 1 i get the following error :

ikram-md commented 4 months ago

For now i cannot use a greater batch size than 1, and when using a batch_size 1 other issues are provoked.

yrcong commented 4 months ago

Please check if you have changed the dataset code.

For the first error (Image 1), it seems you do not pad the images in the same size so that they can be concatenated.

For the second error (Image 2), definitely something wrong with the dataset or collate function. t should be a dict.