search

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Apache License 2.0
334 stars 102 forks source link

PyTorch or TensorFlow #21

Closed Aadyant12 closed 2 years ago

Aadyant12 commented 2 years ago

Can a new example be built on the TensorFlow framework?

Aadyant12 commented 2 years ago

More specifically I have TensorFlow based deep neural network model. How should I proceed if I want to upload it as an example?

FrankLeeeee commented 2 years ago

Hi, we currently do not support TensorFlow even though it is in our future plan. Currently, you need to implement your model in PyTorch and create an example.