Advertisement

PyTorch IndexError: 超出张量维度为0的索引数量

阅读量：

在使用pytorch框架搭建CNN模型的过程中，出现了以下问题：
IndexError: 张量维度不足，索引数量过多
相关代码示例如下：

复制代码 
    JitterTest_Dataset = JitterDataset(
    csv_file= './jitter_labels/new_jitter_test.CSV',
    root_dir= './Test_jitter_Square/',
    transform= transforms.Compose([
        transforms.Grayscale(num_output_channels=1),
        transforms.ToTensor(),
        transforms.Normalize((0.5),(0.5,)),
    )

具体存在的问题为：

复制代码 
    transforms.Normalize((0.5),(0.5,)),

需要在第一个0.5之后添加逗号。
问题分析：
这可能与所使用的pytorch版本有关。
将代码修改为：

复制代码

全部评论 (0)

还没有任何评论哟~

相关文章推荐