This commit is contained in:
Xueyun Zhu 2020-07-15 12:05:24 -07:00 committed by GitHub
parent 0ebe2fab51
commit 5f188f4cf4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1510,7 +1510,7 @@ TEST(GradientGraphBuilderTest, TrainingSession_WithPipeline) {
std::vector<PathString> sub_model_files(num_subs);
for (size_t sub_id = 0; sub_id < num_subs; ++sub_id) {
sub_model_files[sub_id] = GenerateFileNameWithIndex(ORT_TSTR("sub_"), sub_id, ORT_TSTR(".onnx"));
sub_model_files[sub_id] = GenerateFileNameWithIndex(ORT_TSTR("sub_"), static_cast<int>(sub_id), ORT_TSTR(".onnx"));
}
PipelineSplitter splitter;