Learn more about cloning repositories
You have read-only access
Add batch-size to training-metod and add plot-method for illustrating U-net architecture.