hli1221 / imagefusion_densefuse

DenseFuse (IEEE TIP 2019, Highly Cited Paper) - Python 3.6, TensorFlow 1.8.0
244 stars 62 forks source link

训练时长 #6

Closed kourong closed 4 years ago

hli1221 commented 5 years ago

你好,按照论文中的训练设置(79000张图片,batch_size=2,epoch=4),在GTX1080Ti上训练,几个小时就可以训练完成。 训练好的模型在models中的‘zip’压缩文件中。