CommissarMa / Context-Aware_Crowd_Counting-pytorch

The implementation of Context-Aware Crowd Counting(CVPR2019)
MIT License
68 stars 23 forks source link

训练过程大概需要多久呢?SHTech_PartA #5

Closed chw9413 closed 5 years ago

chw9413 commented 5 years ago

第一次接触这类程序,请不要介意我的小白问题。.npy文件已成功生成,显卡是gtx1080,直接跑的代码里的数据集。在训练过程,每次读满两个进度条之后就出现[Errno 111] Connection refused的错误,然后又重新开始读两个进度条。不知道是我哪里的设置有问题,还是说本来这个训练的过程就需要一天以上的时间。 1571834256

yanfangfangfang commented 5 years ago

这个错误应该是 visdom 这个库的原因 我也有报错,我注释了这个部分就没有报错了。但是我也不知道怎么改,如果你知道了 麻烦告诉我一下 谢谢

chw9413 commented 5 years ago

这个错误应该是 visdom 这个库的原因 我也有报错,我注释了这个部分就没有报错了。但是我也不知道怎么改,如果你知道了 麻烦告诉我一下 谢谢

那你注释掉那个库的话对结果有什么影响吗?还有就是你训练花了多久时间呢?

CommissarMa commented 5 years ago

@chw9413 @yanfangfangfang 我更新了readme,visdom是一个可视化库,类似tensorboard,使用前需要在命令行中输入python -m visdom.server,这样就可以在浏览器中看到输出的密度图了。

CommissarMa commented 5 years ago

@chw9413 @yanfangfangfang 训练时间取决于你训练多少个epoch,基本上一个epoch在1分钟左右。

chw9413 commented 5 years ago

@CommissarMa 噢噢,好的,谢谢大佬解答。默认的epoch得到20000是吧?一天过去了,我才到209…我还是改小一点吧…再把visdom加上

chw9413 commented 5 years ago

@aain3314 就跑一下data_preparation里面的代码就可以了。要等一段时间就能看到npy文件了