在自己的数据集上训练时出错:张量大小不匹配
作者: sw5park创建于 2021年3月14日更新于 2024年7月10日
我正在使用自己的数据集、一组图像和对应的标签图进行训练。 我使用了以下命令：ran Python train.py --name clayseam --batchSize 4 --label_nc 2 --no_instance --dataroot /pub2/sw5park/p2phd，其中dataroot文件夹包含4个文件夹：train_label、train_img、test_label、test_img。 但运行时出现以下错误：
内容来源: NVIDIA/pix2pixHD