Enhancement: Train model on GPU

swghosh / DeepFace

Keras implementation of the renowned publication "DeepFace: Closing the Gap to Human-Level Performance in Face Verification" by Taigman et al. Pre-trained weights on VGGFace2 dataset.

https://research.fb.com/publications/deepface-closing-the-gap-to-human-level-performance-in-face-verification/

MIT License

194 stars 61 forks source link

Enhancement: Train model on GPU #3

Closed ajinkya933 closed 4 years ago

ajinkya933 commented 4 years ago

I have made changes to the TPU code and have made it GPU compatible