amzn / trans-encoder

Trans-Encoder: Unsupervised sentence-pair modelling through self- and mutual-distillations
Apache License 2.0
133 stars 16 forks source link

Run code without CUDA #9

Open KIlian42 opened 1 year ago

KIlian42 commented 1 year ago

First of all, great paper and great that you guys make your code publish available. :-)

For both I need CUDA: train_self_distill.sh and train_mutual_distill.sh. Does the repo/training also works without CUDA available?

hardyqr commented 1 year ago

Hey, thanks for your interest :) Unfortunately, I didn't explicitly implement the situation when CUDA/GPU is unavailable. I imagine one needs to tweak the code a bit to make this possible.