Commit Graph

4 Commits

Author SHA1 Message Date
Subuday 5fd7ea93ea Add upsampling and downsampling to UNet 2024-02-15 13:24:30 +00:00
Subuday fd6c0afbbf Add ResNetBlock1D to UNet 2024-02-15 08:40:04 +00:00
Subuday b5467b8051 Add UNet backbone 2024-02-12 21:44:29 +00:00
Subuday 7314b1cbec Implement model forward 2024-02-12 19:39:22 +00:00