byronknoll/cmix

Question about GPU utilization

dillfrescott opened this issue · 2 comments

dillfrescott commented

Would it be possible to utilize an nvidia gpu for pretraining?

byronknoll commented

It is possible, but it is more appropriate for a separate project rather than cmix. e.g. nncp and tensorflow-compress use GPU.

  • 👍1
dillfrescott commented

Ah, makes sense!