摘要: class semiDataset(Dataset): def __init__(self, no_label_loder, model, device, thres=0.99): x, y = self.get_label(no_label_loder, model, device, thres) 阅读全文
posted @ 2025-02-01 11:20 JYP0222 阅读(26) 评论(0) 推荐(0)
摘要: 随机种子固定随机结果,方便复现 def seed_everything(seed): torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) torch.backends.cudnn. 阅读全文
posted @ 2025-02-01 09:51 JYP0222 阅读(39) 评论(0) 推荐(0)