小小概念之你能奈我何
文章平均质量分 76
深度学习系列
就要酸死你
这个作者很懒,什么都没留下…
展开
-
pytorch中model.train(),model.eval()和torch.no_grad()的区别
使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval。原创 2022-09-15 22:11:49 · 279 阅读 · 0 评论 -
pytorch之加载数据集Dataset and DataLoader
举例:假设有一个包含10000个样本的数据集,并且您选择的Batch Size为1000和100个Epoch。这意味着数据集将分为10个Batch,有10个iterations,每个Batch有1000个样本。假设样本总数是87, Batchsize是8, 如果drop_last=True, 那么1Epoch=10Iteration, 如果等于False, 那么1Epoch=11Iteration, 最后1个Iteration有7个样本。Batch Size(批大小):每批样本的大小。是否多进程读取机制,原创 2022-09-15 22:10:36 · 412 阅读 · 0 评论