我可以使用float16使用ConvNet对图像进行去噪吗?

2024-04-25 03:54:04 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在用一个大修道院用一些大的卷积神经网络去噪图像。我使用的是使用T4 GPU的谷歌云。我读到使用Float16可以显著提高训练速度。我尝试将模型和数据转换为float16,但当我在训练期间使用测试数据集的单个样本来查看去噪过程时,matplotlib.plt只会返回一个黑色图像(再次转换为float32后)。而且,培训似乎不起作用。我想知道,在理论中,使用float16是否有效,如何才能恢复真实的图像


Tags: 数据模型图像gpumatplotlib过程神经网络卷积