RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 756 and 1008 in dimension 2 原因:輸入的圖像數據的維度不完全一樣(數據集圖像大小不一)。 解決方法: transforms.Compose([ transforms.RandomResizedC
相關文章
相關標籤/搜索