This repository contains 2D and 3D U-Net TensorFlow scripts for training models using the Medical Decathlon dataset (http://medicaldecathlon.com/).
David Ojika, Bhavesh Patel, G. Athony Reina, Trent Boyer, Chad Martin and Prashant Shah. “Addressing the Memory Bottleneck in AI Model Training”, Workshop on MLOps Systems, Austin TX (2020) held in conjunction with Third Conference on Machine Learning and Systems (MLSys). https://arxiv.org/abs/2003.08732