Commit Graph

1 Commits

Author SHA1 Message Date
Erin cc2cc00d20
Add tests for 2D UNet blocks (#1945)
* test unet blocks 2d

* change to randn_tensor

* mps flaky
2023-01-16 12:53:05 +01:00