mosaicml / examples

Fast and flexible reference benchmarks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

will fine tuning work on windows 11 lap top minimal gpu for bert sequence_classification.py ?

Sandy4321 opened this issue · comments

https://github.com/mosaicml/examples/blob/main/examples/benchmarks/bert/sequence_classification.py

will fine tuning work on windows 11 lap top minimal gpu ?
print(torch.cuda.get_device_name(0))
NVIDIA GeForce RTX 3050 Ti Laptop GPU

I believe it should work since its an NVIDIA GPU. We haven't tested it, so please let us know if you hit any issues. Happy to help!

I'm going to close for now, but feel free to reopen if you hit issues

sure
what is link to code example to try for text binary classificatoin ?

Take a look at the BERT examples: https://github.com/mosaicml/examples/tree/main/examples/benchmarks/bert. Let me know if you have more questions after

great thanks for soon answer
then what munimal computing I Need for classification for fine tuning IMDB data set?
1 is it enouth CPU only laptop 32 GB RAM?
2 will it run on free google colab tier ?
2 will it run on free AWS sagemaker lab ?

Unfortunately, we don't have those kind of details on hand. I recommend trying it and seeing what happens!