文件名称:Tensorflow-Multi-GPU-Training.rar
文件大小:31.81MB
文件格式:RAR
更新时间:2023-08-18 04:37:54
Tensorflow多GPU
使用Tensorflow数据并行,多GPU训练的demo。模型是一个image2image的模型,参考了dncnn。训练数据转为TFRecord文件。
【文件预览】:
Tensorflow-Multi-GPU-Training
----train.py(7KB)
----test.py(1KB)
----utils.py(3KB)
----dataset()
--------gt()
--------TFRecord()
--------noisy()
----network.py(993B)
----checkpoint()
--------denoising.ckpt.index(6KB)
--------denoising.ckpt.meta(1.94MB)
--------README.md(124B)
--------denoising.ckpt.data-00000-of-00001(1.45MB)
--------checkpoint(85B)