- 博客(11)
- 收藏
- 关注
原创 pytorch中model.train(),model.eval()和torch.no_grad()的区别
使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval。
2022-09-15 22:11:49 261
原创 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 376
原创 pytorch-lighting使用
其中model为实例化的pl.LightningModule。也就是定义model类是会定义哪些函数。开始训练前加载n个验证数据进行测试,添加回调函数或回调函数列表。k=-1时加载所有验证数据。
2022-09-14 11:44:20 1685 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人