IndexError: index 1 is out of bounds for axis 0 with size 1

#仅供自己参考

该错误是索引超出了列表的长度的,比如创建了长度为1的数组a,而我的索引为在a[1]:

import numpy as np
a = np.empty(1)
print(a[1])

就会报错:

IndexError: index 1 is out of bounds for axis 0 with size 1

所以这时候就回去检查是自己的索引错了, 还是数组长度定义错了。

原文链接:https://blog.csdn.net/weixin_44493244/article/details/105968388

然而我在尝试了以上方法后,发现并不能解决问题,所以我的问题不是出在这,因此仔细检查源代码发现,

 for i in pbar:
            #convert from a torch.Size object to a list
            if self.verbose: pbar.set_description("buffering images")

            datum_index = shuffed_data_order[i]
            datum = dataset[datum_index]
            
            datum_size = datum[0].size()
            
            self.buffer_history.append(datum_index)
            self.buffer.append(datum)
            
        self.remaining_to_be_in_buffer = shuffed_data_order[i+1:]
            
        self.patch_size = [datum_size[0]] + patch_size

与buffer_size有关,将buffer_size设置为1运行,问题解决。

DATASET=${1:-LMmed}
BUFFER_SIZE=1 #originally 8, patch_size origianlly 256 256
N_ITER=10
RUN_DIR="saved_models/${DATASET}"
PATH_DATASET_ALL_CSV="data/csvs/${DATASET}.csv"
PATH_DATASET_TRAIN_CSV="data/csvs/${DATASET}/train.csv"
GPU_IDS=${2:-0}

你可能感兴趣的:(numpy)