摘要: train.py的221行是做train_loader。在make_dataloader中关键代码是dataset = build_dataset(cfg, is_train)然后直接data_loader = torch.utils.data.DataLoader。 而build_dataset又 阅读全文
posted @ 2023-01-13 22:29 祥瑞哈哈哈 阅读(105) 评论(0) 推荐(0)