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