train_generic only accepts model_fn....

would be cool to have a config that accepts an Estimator as input

To upload designs, you'll need to enable LFS and have an admin enable hashed storage. More information