Train a u-net for infilling data

[HastingsGreer]

This is the first pass at porting my u-net code to this codebase. Next step is to visualize results, and use it to implement a policy