摘要: 问题起源于为了完成一个多任务的训练,自己写了个 BatchSampler,如下: class MultiTaskBatchSampler(BatchSampler): def __init__(self, datasets: MultiTaskDataset, batch_size: int, sh 阅读全文
posted @ 2020-12-11 17:42 GNEPIUHUX 阅读(129) 评论(0) 推荐(0) 编辑