Get a numpy array of weights, one per diffusion step. The weights needn't be normalized, but must be positive.
(self)
| 33 | |
| 34 | @abstractmethod |
| 35 | def weights(self): |
| 36 | """ |
| 37 | Get a numpy array of weights, one per diffusion step. |
| 38 | |
| 39 | The weights needn't be normalized, but must be positive. |
| 40 | """ |
| 41 | |
| 42 | def sample(self, batch_size, device): |
| 43 | """ |