Giters

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch or TensorFlow

Aadyant12 opened this issue · comments

commented

Can a new example be built on the TensorFlow framework?

commented

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

commented

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.