excessive memory usage when training svm

fjchange / object_centric_VAD

An Tensorflow Re-Implement of CVPR 2019 "Object-centric Auto-Encoders and Dummy Anomalies for Abnormal Event Detection in Video"

MIT License

97 stars 30 forks source link

excessive memory usage when training svm #3

Closed xiadingZ closed 5 years ago

xiadingZ commented 5 years ago

I implement in pytorch and follow your settings and get 61.9 auc on shanghaitech, using only rgb. can you show your auc of it? If I use three-stream, training of svm will terminate because of excessive memory usage, on my server. I use pytorch's torchvision fasterrcnn_resnet50_fpn model to detect object, what's your number of objects on shanghaitech? Thanks very much

fjchange commented 5 years ago

I don't know what is your training param setting, but mine is showed on #2 . The features is too large to saved in .m with scipy.io for ShanghaiTech dataset, so I haven't use it to train on matlab. The number of objects I got on ShanghaiTech is about 190000.