How to use tensorflow-workspace to train my own model?