admin管理员组

文章数量:1558091

torch训练卡死,后来发现是

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

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

本文标签: 进程torch