age-gender-estimation icon indicating copy to clipboard operation
age-gender-estimation copied to clipboard

Published 20 hours ago • verified publisher icon yu4u

Train

Open longlongvip opened this issue 5 years ago • 3 comments

I have a PC with RTX2060 GPU,but it use the CPU when I train the network,I have installed the tensorflow-gpu

longlongvip avatar Jul 05 '19 09:07 longlongvip

Did you try Keras examples to check whether GPU is used or not?

yu4u avatar Jul 07 '19 15:07 yu4u

I had same issue. Updating Nvidia Drivers fixed the issue for me.

lkdmttg7 avatar Nov 13 '19 12:11 lkdmttg7

@yu4u When i train by 【python train.py --input data/imdb_db.mat】，result：all face test ”49 M“。Do you need to load your pre-training model during training？

troublemaker-r avatar Jun 16 '20 01:06 troublemaker-r