make grid 的 normalize=True 对单张图不生效
make grid 的 normalize=True 对单张图不生效: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). make grid 的 normalize=True 对单张图不生效 我在用 jittor 训一个生成模型，每隔若干步把当前生成出来的图存盘看一下效果。生成出来的张量数值范围不是 0, 1 （大致在 -1, 1 左右），所以我用 make grid / save image 时传了 normalize=True，希望它在保存前把数值缩放到 0, 1 。 batch…
The task
make_grid 的 normalize=True 对单张图不生效 我在用 jittor 训一个生成模型，每隔若干步把当前生成出来的图存盘看一下效果。生成出来的张量数值范围不是 [0, 1]（大致在 [-1, 1] 左右），所以我用 make_grid / save_image 时传了 normalize=True，希望它在保存前把数值缩放到 [0, 1]。 batch 内有多张图时一切正常，存出来的拼图颜色看上去是对的。 但是当我只想看单张图的时候（比如只生成一个样本，或者 debug 阶段只 forward…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.