Closed eshedmargalit closed 6 years ago
Modifies base to deal with case where model_fn outputs are a dict
Modifies base to deal with case where model_fn outputs are a dict