One GPU is working during Evaluation

zhongyy / Face-Transformer

Face Transformer for Recognition

MIT License

252 stars 54 forks source link

One GPU is working during Evaluation #15

Open khawar-islam opened 3 years ago

khawar-islam commented 3 years ago

Hi @zhongyy

I have noted that while training all GPUs are working but when performance evaluation is started only 1 GPU is used. Would it be possible to use four GPUs during performance evaluation? It will make the process faster.

Regards, Khawar

zhongyy commented 3 years ago

Hi Khawar, Sorry for the late reply. Thanks for your interests, and good advices. While it is not easy for me to implement your good ideas. Transformers realy consumes a lot and I am busy with another project these days. So I am afraid that I cannot update it.