I'll be adding more model implementations (mainly CV models) to this repo in the future. This is mainly meant for practicing PyTorch and able to reproduce the model architectures in the papers I come across.
Models added:
- UNet (Paper: https://arxiv.org/abs/1505.04597)