From c25ad11b4312bcd2b65c63ce07fa3d62698af57e Mon Sep 17 00:00:00 2001
From: Victor Bittorf <vbittorf@google.com>
Date: Tue, 22 Jun 2021 15:31:55 -0700
Subject: [PATCH] Changes to Training

---
 mnist/tasks/train.yaml | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/mnist/tasks/train.yaml b/mnist/tasks/train.yaml
index 580e9ae..f55c984 100644
--- a/mnist/tasks/train.yaml
+++ b/mnist/tasks/train.yaml
@@ -9,10 +9,13 @@ inputs:
 
         - name: parameters_file
           type: file
+          
+        - name: model_in
+          type: directory
 # Task Outputs
 outputs:
-        - name: log_dir
-          type: directory
+        - name: metrics
+          type: file
 
         - name: model_dir
           type: directory