如何查看自己的batch——size
时间: 2023-08-02 10:07:58 浏览: 148
【BERT系列】——命名实体识别
5星 · 资源好评率100%
如果你已经在训练深度学习模型,并且使用的是框架如TensorFlow、PyTorch等,那么可以通过以下方式来查看当前的 batch size:
在 TensorFlow 中,可以通过以下代码来查看当前的 batch size:
```python
import tensorflow as tf
batch_size = tf.shape(input_tensor)[0]
```
在 PyTorch 中,可以通过以下代码来查看当前的 batch size:
```python
batch_size = tensor.size(0)
```
其中,input_tensor 或 tensor 是当前模型输入的张量,可以根据实际情况进行修改。如果你在训练模型时没有设置 batch size,那么默认的 batch size 是 1。
阅读全文