This repository was archived by the owner on Sep 10, 2025. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +7
-9
lines changed Expand file tree Collapse file tree 3 files changed +7
-9
lines changed Original file line number Diff line number Diff line change 7474~~~
7575
7676.. autoclass :: SST
77- :members: splits, iters
77+ :members: __init__, splits, iters
7878
7979IMDb
8080~~~~
8181
8282.. autoclass :: IMDB
83- :members: splits, iters
83+ :members: __init__, splits, iters
8484
8585
8686Text Classification
Original file line number Diff line number Diff line change @@ -163,13 +163,9 @@ setup_pip_pytorch_version() {
163163 # Install latest prerelease version of torch, per our nightlies, consistent
164164 # with the requested cuda version
165165 pip_install --pre torch -f " https://download.pytorch.org/whl/nightly/${WHEEL_DIR} torch_nightly.html"
166- if [[ " $CUDA_VERSION " == " cpu" ]]; then
167- # CUDA and CPU are ABI compatible on the CPU-only parts, so strip
168- # in this case
169- export PYTORCH_VERSION=" $( pip show torch | grep ^Version: | sed ' s/Version: *//' | sed ' s/+.\+//' ) "
170- else
171- export PYTORCH_VERSION=" $( pip show torch | grep ^Version: | sed ' s/Version: *//' ) "
172- fi
166+ # CUDA and CPU are ABI compatible on the CPU-only parts, so strip
167+ # in this case
168+ export PYTORCH_VERSION=" $( pip show torch | grep ^Version: | sed ' s/Version: *//' | sed ' s/+.\+//' ) "
173169 else
174170 pip_install " torch==$PYTORCH_VERSION$PYTORCH_VERSION_SUFFIX " \
175171 -f https://download.pytorch.org/whl/torch_stable.html \
Original file line number Diff line number Diff line change @@ -25,6 +25,8 @@ requirements:
2525
2626build :
2727 string : py{{py}}
28+ script_env :
29+ - BUILD_VERSION
2830
2931test :
3032 imports :
You can’t perform that action at this time.
0 commit comments