search

Sierkinhane / mtcnn-pytorch

A face detection algorithm
317 stars 162 forks source link

gen_rnet_data #9

Open Sarah-Elfanty opened 5 years ago

Sarah-Elfanty commented 5 years ago

Hi, I've been trying to train this model from scratch on a local dataset, but I've been encountering a CUDA memory issue when i run gen_Rnet_train_data.py I'm using a 1050ti , 4gb GPU with 16 gb RAM.

Sierkinhane commented 5 years ago

Maybe you should reduce the batch size of testimageloader

liaomuqin commented 4 years ago

我也遇到了同样的问题，请问怎么解决。。。