loader迭代到最后,直接终止
收藏
我运行的是ppseg-release-2.5 ppLiteSeg源码,训练loader和config部分如下:
# ---------------- loader ----------------# loader = ...... iter = start_iter while iter < iters: for data in loader: iter+=1 print(iter,' ----- ',data[0].shape,data[1].shape) ''' 其余操作均注释掉 ''' # ---------------- config -----------------# train_dataset: type: Dataset dataset_root: data/camvid num_classes: 2 mode: train train_path: data/camvid/train.txt transforms: - type: Normalize val_dataset: type: Dataset dataset_root: data/camvid num_classes: 2 mode: val val_path: data/camvid/val.txt transforms: - type: Normalize
(1)win10
(2)我的图像尺寸全部一致(3,1024,256),标签(1024,256),分割目标一个类别;
(3)数据处理操作全部去掉,仅仅保留归一化;
(4)不管使用多少张图像,不管val部分是否关掉,都是迭代完一次 loader 后终止,不跳到下次loader迭代
比如:
iters=20000、1000张图、bs=2:输出 500,' ----- ',[2,3,256,1024],[2,256,1024],然后终止
iters=20000、10张图、bs=2:输出 5,' ----- ',[2,3,256,1024],[2,256,1024],然后终止
请问有人碰到过这个问题吗?
0
收藏
请登录后评论