使用CIFAR-10数据集STUCK以低精度进行简单的CNN模型训练

作者: 猫南北
发布时间: 2024-12-17 07:11:08 (2月前)
转自：

2 条回复

0#
回复此人
1号 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 由于批量大小为1，因此使用较低的学习率（如1e-4）或增加批量大小。 </p> <P> 我建议批量大小为16或更大。 </p> <P> 编辑：要创建一批数据，您可以执行以下操作。 </p> <pre> <code> N = input.shape[0] #know the total size/samples in input for i in range(n_epochs): # this is to shuffle data indices = torch.randperm(N) for idx in range(0, N, batch_size): batch_input = input[idx:idx+batch_size] # this will get you input of size batch_size # do whatever you want with the batch_input # .... </code> </pre> </DIV>

编辑

登录后才能参与评论