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!