torch 多进程卡死

torch训练卡死,后来发现是

DataLoader卡死,原因是enumerate参数不对,

dataloader = DataLoader(COCODataset(config["train_path"],
                                                 (config["img_w"], config["img_h"]),
                                                 is_training=True, data_size=100000),
batch_size=config["batch_size"], shuffle=True,drop_las

版权声明:本文为jacke121原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。