7 Commits

Author	SHA1	Message	Date
Subuday	f15230bb67	Add transformer block to UNet	2024-02-15 18:52:42 +00:00
Subuday	5fd7ea93ea	Add upsampling and downsampling to UNet	2024-02-15 13:24:30 +00:00
Subuday	8676ab30d9	Fix appending a new block to input_blocks	2024-02-15 08:55:52 +00:00
Subuday	fd6c0afbbf	Add ResNetBlock1D to UNet	2024-02-15 08:40:04 +00:00
Subuday	0f7a7edb9b	Add conv block to UNet	2024-02-14 21:21:07 +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