(self, data, t)
| 422 | |
| 423 | |
| 424 | def _denoise(self, data, t): |
| 425 | B, D,N= data.shape |
| 426 | assert data.dtype == torch.float |
| 427 | assert t.shape == torch.Size([B]) and t.dtype == torch.int64 |
| 428 | |
| 429 | out = self.model(data, t) |
| 430 | |
| 431 | assert out.shape == torch.Size([B, D, N]) |
| 432 | return out |
| 433 | |
| 434 | def get_loss_iter(self, data, noises=None): |
| 435 | B, D, N = data.shape |
nothing calls this directly
no outgoing calls
no test coverage detected