2023年3月16日
optimizer.zero_grad()
摘要： 传统的训练函数，一个batch是这么训练的： for i,(images,target) in enumerate(train_loader): # 1. input output images = images.cuda(non_blocking=True) target = torch.from 阅读全文
posted @ 2023-03-16 17:15 sqsq 阅读(137) 评论(0) 推荐(0) 编辑