mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
fe50213491
* bert single node multi GPU pretrain w/o checkpoint Co-authored-by: liqun <liqun@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
9 lines
227 B
Text
9 lines
227 B
Text
--pre
|
|
-f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
|
|
# transformers requires sklearn
|
|
sklearn
|
|
transformers==v2.10.0
|
|
torch==1.6.0.dev20200610
|
|
torchvision==0.7.0.dev20200610
|
|
torchtext==0.6.0.dev20200610
|
|
h5py
|