RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0 #154

duanqipeng · 2020-01-13T09:18:05Z

i got a problem with
RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0

use four gpu

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0 #154

RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0 #154

duanqipeng commented Jan 13, 2020

RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0 #154

RuntimeError: The shape of the mask [2, 128, 128] at index 0 does not match the shape of the indexed tensor [1, 2, 2, 128] at index 0 #154

Comments

duanqipeng commented Jan 13, 2020