4 Commits

Author	SHA1	Message	Date
Yin Li	7b6ff73be1	Move UNet with ResBlock to VNet and Revert UNet to the previous simple version	2019-12-09 21:53:27 -05:00
Yin Li	0764a1006e	Remove unnecessary arguments --in-channels and --out-channels	2019-12-09 10:19:21 -05:00
Yin Li	11c9caa1e2	Fix unstable training by limiting pytorch version to 1.1	2019-12-08 21:02:08 -05:00
Yin Li	0211eed0ec	Add testing	2019-12-01 21:40:35 -05:00