(self, data, t)
| 293 | |
| 294 | |
| 295 | def _denoise(self, data, t): |
| 296 | B, D,N= data.shape |
| 297 | assert data.dtype == torch.float |
| 298 | assert t.shape == torch.Size([B]) and t.dtype == torch.int64 |
| 299 | |
| 300 | out = self.model(data, t) |
| 301 | |
| 302 | assert out.shape == torch.Size([B, D, N]) |
| 303 | return out |
| 304 | |
| 305 | def get_loss_iter(self, data, noises=None): |
| 306 | B, D, N = data.shape |
nothing calls this directly
no outgoing calls
no test coverage detected