Skip to content

prog_bar = tqdm(enumerate(train_data_loader))这一行卡住的话,是因为filelists/train.txt, filelists/val.txt, filelists/test.txt没配置好,导致Dataset死循环 #6

@yuanmaitian

Description

@yuanmaitian

prog_bar = tqdm(enumerate(train_data_loader)) 这一行卡住的话,是因为filelists/train.txt, filelists/val.txt, filelists/test.txt没配置好,导致Dataset死循环
现在我也遇到了这个情况,训练集、验证集的全路径写在了trian.txt和val.txt ,类似下面的:
/root/workspace/processed_data/train/57f7d1d11ca308f3f24728cb5c930574_15.78_16.16
/root/workspace/processed_data/train/4c3df35b8ffdbd064ce3d00288c0d26a_2.44_2.71
/root/workspace/processed_data/train/f5a9b275800eeaa1561b72f5b412908c_2.88_3.3

/root/workspace/processed_data/val/01b7a9520196fad67d4670fdf85cf59d_1.08_1.32
/root/workspace/processed_data/val/8b57ece4a8a0a07395ec833d712497f6_4.68_5.22
/root/workspace/processed_data/val/ade95c1b7497724a5201de58c7304f59_1.6_1.94
然后说原版的代码处理数据集会死循环,要怎么修改呢?请假大佬

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions