DreamerDoReMi

训练速度很慢

6

您好，我对您的论文很感兴趣， 现在在尝试跑您这份代码，但是loss.backword()的时候大概每张图需要0.6s, batch_size 设置为1时，每个iteration需要6s左右，我的显卡是1080Ti，系统是Ubuntu16.04. 请问您在训练的时候速度也是这么慢吗？

Excuse me, could you show me some hints about how to test on my own dataset since you had set the segmentation label for all your training and testing images?...