search

yu4u / age-gender-estimation

Keras implementation of a CNN network for age and gender estimation
MIT License
1.47k stars 503 forks source link

Train #95

Open longlongvip opened 5 years ago

longlongvip commented 5 years ago

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

yu4u commented 5 years ago

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

lkdmttg7 commented 4 years ago

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

troublemaker-r commented 4 years ago

@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？