Get a numpy array of weights, one per diffusion step. The weights needn't be normalized, but must be positive.
(self)
| 39 | |
| 40 | @abstractmethod |
| 41 | def weights(self): |
| 42 | """ |
| 43 | Get a numpy array of weights, one per diffusion step. |
| 44 | The weights needn't be normalized, but must be positive. |
| 45 | """ |
| 46 | |
| 47 | def sample(self, batch_size, device): |
| 48 | """ |