[在训练rasa-nlu时，我遇到ResourceExhaustedError。所以我使用了固定的批量大小。但是由于我将拥有可变大小的数据集，因此我需要知道如何为任何给定的数据集计算最大的批处理大小。有什么办法吗？
我知道可以使用以下公式计算最大批量：
最大批大小=可用的GPU内存字节/ 4 /（张量的大小+可训练的参数）
我如何获得张量大小和可训练参数的数量？
请参见https://forum.rasa.com/t/determine-the-largest-possible-batch-size/25864，以解决此问题，因为它正在那里解决