DushyantaDhyani / kdtf

Knowledge Distillation using Tensorflow
MIT License
141 stars 47 forks source link

multi-gpus #5

Open zdgithub opened 4 years ago

zdgithub commented 4 years ago

Could the distillation networks be trained on multiple gpus?

DushyantaDhyani commented 4 years ago

That would be definitely worth trying. Don't have the bandwidth to work on that though