pytorch中训练神经网络时如何计算batch数

在 PyTorch 中，可以通过以下方式计算 batch 数量：

batch_size = 32 # 假设 batch size 是 32
total_samples = len(dataset) # dataset 是你的数据集对象，len(dataset) 返回数据集的总样本数
num_batches = total_samples // batch_size # // 表示整除，计算整个数据集可以分成几个 batch
if total_samples % batch_size != 0: # 如果不能整除，还需要再加上一个 batch
    num_batches += 1

这样，`num_batches` 就是数据集分成的 batch 数量了。在训练过程中，每次迭代都会取出一个 batch 的数据进行训练。