hpcaitech / ColossalAI-Examples

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

Examples of Generative Adversarial Networks #89

Closed jnbai517 closed 2 years ago

jnbai517 commented 2 years ago

Add a new example based on AC-GAN.

binmakeswell commented 2 years ago

Hi @bajiaoyu517 Thanks for your contribution! Here are some suggestions for improvement:

  1. Upload the file to /image/ rather than /features/.
  2. Polish README: It seems copy from other place; Why you provide this example to Colossal-AI, how this example will help others, what is the expectation training result for this example, and did you verify its correctness (please provide the satisfactory log)?
  3. Currently, this example does not support main features in Colossal-AI, eg. data/tensor/pipeline parallel, ZeRO, etc. Could you please try to support them, provide the configs, and update expected results in README? Thanks.
binmakeswell commented 2 years ago

This PR is closed due to inactivity.