saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-07 00:13:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

For orttraining

Browse source
This commit is contained in:
Ryan Hill 2021-05-10 16:18:56 -07:00
parent a59cf4b050
commit 149c037a4a
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
orttraining/orttraining/models/bert/main.cc
View file

@ -616,6 +616,7 @@ void setup_training_params(BertParameters& params) {
0,
true,
0,
nullptr,
nullptr};
if (params.gpu_mem_limit_in_gb > 0) {