Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation

This repo contains an official PyTorch implementation for the paper Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation.

We propose a genearlly applicable training method for a general weighted diffusion loss.

To-Do

• Release the code
• Release the running commands
• Release the checkpoints and samples