search

feiyuhuahuo / Anomaly_Prediction

Pytorch implementation of anomaly prediction.
78 stars 22 forks source link

train.py #27

Closed qiangmao306 closed 2 years ago

qiangmao306 commented 3 years ago

我在训练时卡在dataloader循环之前。我注意到你在另一个工作 pre_ano_detection那边也有相同的问题，程序卡住下不去，你的这份代码我也遇到了同样的问题，请问你知道原因吗 image

feiyuhuahuo commented 3 years ago

查查模型能不能顺利读到数据

youngupup commented 3 years ago

@feiyuhuahuo ，请问您训练并测试过shanghai dataset吗？我用自己的数据集测试auc不是很理想。感谢您的工作

feiyuhuahuo commented 3 years ago

shanghaitech 太大了，没训练。这个模型的应用场景有限，只能用在规律运动场景，类似商场人流那种。

youngupup commented 3 years ago

shanghaitech 太大了，没训练。这个模型的应用场景有限，只能用在规律运动场景，类似商场人流那种。

好的，感谢