#仅供自己参考
该错误是索引超出了列表的长度的,比如创建了长度为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}