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训练卡死,后来发现是
DataLoader卡死,原因是enumerate参数不对,
dataloader = DataLoader(COCODataset(config["train_path"], (config["img_w"], config["img_h"]), is_training=True, data_size=100000), batch_size=config[