Commit Graph

3 Commits

Author SHA1 Message Date
3530d91aaf
fix: add dropout layer into unet model 2023-03-14 18:18:13 +08:00
7a0b5a2a8a
fix: remove nn.Tanh() on the last layer 2023-03-14 02:14:19 +08:00
9373320e4a
feat: Unet module 2023-03-13 21:29:16 +08:00