#python #pytorch
Вопрос:
Когда я хочу сохранить сгенерированную маску сегментации с помощью функции save_image . Размер сегментации (4x256x256) сложен. Код сохранения-это
save_image(final_images, "./output/results_page_{}.png".format(count), nrow=2*opt.num_variations 1, padding=2, range=(0, 1), pad_value=0.5, normalize=False)
На сервере с CUDA 11.0 и PyTorch 1.7.1 py3.7_cuda11.0.221_cudnn8.0.5_0. Код работает хорошо. И я могу сохранять файлы формата RGBA.
Однако, когда я запускаю код в CUDA 11.1 и PyTorch 1.9.1 py3.7_cuda11.1_cudnn8.0.5_0, он не работает, и вывод
The size of tensor a (3) must match the size of tensor b (4) at non-singleton dimension 0
.Поэтому мне нужно преобразовать маску сегментации convert("RGB") в изображение RBG и сохранить его.
Итак, есть ли разница между этими двумя ВЕРСИЯМИ?
Комментарии:
1. Это не имеет никакого отношения к CUDA. Я соответствующим образом отредактировал ваш вопрос
2. «Итак, есть разница между этими двумя ВЕРСИЯМИ?» — Я очень надеюсь на это. В противном случае я зря обновил свой PyTorch.
3. Спасибо за исправление