RandomMixUpV2 / RandomCutMixV2 将类标签转换为图像 dtype: bfloat16 会损坏大于 256 的标签

摘要

RandomMixUpV2 和 RandomCutMixV2 将整数类标签转换为 图像 dtype。在 bfloat16 (以及 2048 以上的 float16) 中，标签本身会被四舍五入，因此例如 ImageNet-1k 的类 ID 会出错：