bumped version of tf-wide-deep image to 1.3 in test manifest, and removed the data download from the tf-wide-deep pod spec command

This commit is contained in:
Brian McQueen
2022-10-23 10:04:07 -07:00
parent 7423cc931f
commit 08c22d6d9a
2 changed files with 2 additions and 2 deletions

View File

@@ -56,7 +56,7 @@ func (w tfWideDeepWorkload) PodSpec() v1.PodSpec {
},
},
Command: []string{"/bin/sh"},
Args: []string{"-c", "python ./data_download.py && time -p python ./wide_deep.py --model_type=wide_deep --train_epochs=300 --epochs_between_evals=300 --batch_size=32561"},
Args: []string{"-c", "time -p python ./wide_deep.py --model_type=wide_deep --train_epochs=300 --epochs_between_evals=300 --batch_size=32561"},
}
containers = append(containers, ctn)