首页 Paddle框架 帖子详情
loader迭代到最后,直接终止
收藏
快速回复
Paddle框架 问答模型训练 347 0
loader迭代到最后,直接终止
收藏
快速回复
Paddle框架 问答模型训练 347 0

我运行的是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
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户