Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Allow setting WorkspaceMode for transfer learning as well #4090
What changes were proposed in this pull request?
Add trainingWorkspaceMode and inferenceWorkspaceMode fields to FineTuneConfiguration and use them for the NeuralNetConfiguration.
How was this patch tested?
Builds and tests pass, and manually checked that TransferLearning.GraphBuilder() returns the modified configuration with given trainingWorkspaceMode and inferenceWorkspaceMode.