• Nathan Lambert's avatar
    VE/VP SDE updates (#90) · 63c68d97
    Nathan Lambert authored
    
    
    * improve comments for sde_ve scheduler, init tests
    
    * more comments, tweaking pipelines
    
    * timesteps --> num_training_timesteps, some comments
    
    * merge cpu test, add m1 data
    
    * fix scheduler tests with num_train_timesteps
    
    * make np compatible, add tests for sde ve
    
    * minor default variable fixes
    
    * make style and fix-copies
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    63c68d97
scheduling_pndm.py 8.86 KB