PyTorch के साथ कुशल AI मॉडल प्रशिक्षण
Dennis Lee
Data Engineer, Amazon
$$


$$


$$


$$


$$
$$
$$





$$

$$

$$

$$






AutoModelForSequenceClassification कॉल करके पैरामीटर इनिशियलाइज़ करेंfrom transformers import AutoModelForSequenceClassification model = AutoModelForSequenceClassification.from_pretrained(model_name)print(model.config)
DistilBertConfig {
"architectures": ["DistilBertForMaskedLM"],
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
...
Accelerator हमारे कंप्यूटर पर उपलब्ध डिवाइस पहचानता हैaccelerator.prepare()torch.nn.Module टाइप) पहले उपलब्ध GPU पर रखेंfrom accelerate import Accelerator accelerator = Accelerator() model = accelerator.prepare(model)print(accelerator.device)
cpu
PyTorch के साथ कुशल AI मॉडल प्रशिक्षण