Learn more about cloning repositories
You have read-only access
Add training and predictions-methods to class for U-net.