PyTorch ile Verimli AI Model Eğitimi
Dennis Lee
Data Engineer, Amazon
$$


$$


$$


$$


$$
$$
$$





$$

$$

$$

$$






AutoModelForSequenceClassification çağırarak model parametrelerini başlatınfrom transformers import AutoModelForSequenceClassification model = AutoModelForSequenceClassification.from_pretrained(model_name)print(model.config)
DistilBertConfig {
"architectures": ["DistilBertForMaskedLM"],
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
...
Accelerator bilgisayarımızda hangi cihazların olduğunu algılaraccelerator.prepare()torch.nn.Module türünde) ilk uygun GPU'ya yerleştirirfrom accelerate import Accelerator accelerator = Accelerator() model = accelerator.prepare(model)print(accelerator.device)
cpu
PyTorch ile Verimli AI Model Eğitimi