• Denis's avatar
    Training to predict x0 in training example (#1031) · cbcd0512
    Denis authored
    
    
    * changed training example to add option to train model that predicts x0 (instead of eps), changed DDPM pipeline accordingly
    
    * Revert "changed training example to add option to train model that predicts x0 (instead of eps), changed DDPM pipeline accordingly"
    
    This reverts commit c5efb525648885f2e7df71f4483a9f248515ad61.
    
    * changed training example to add option to train model that predicts x0 (instead of eps), changed DDPM pipeline accordingly
    
    * fixed code style
    Co-authored-by: default avatarlukovnikov <lukovnikov@users.noreply.github.com>
    cbcd0512
train_unconditional.py 16 KB