Trainer सेट करें
आपका मॉडल जटिल वाक्यांशों को सरल और ज्यादा सुलभ पैराफ्रेज़ से बदल देगा ताकि अनुवादों की पठनीयता बेहतर हो. उदाहरण के लिए, यह मूल वाक्यांश "The inclement weather conditions precipitated the postponement of the outdoor event" को सरल करके "The bad weather caused the outdoor event to be delayed." कर सकता है. अपने भाषा अनुवाद सर्विस की ट्रेनिंग की तैयारी के लिए Trainer बनाइए! trainer.train() कॉल होने के कारण इस अभ्यास को चलने में कुछ समय लगेगा.
कुछ डेटा पहले से लोड है:
modelएक Transformer मॉडल हैdatasetमें वाक्य पैराफ्रेज़ का MRPC डेटासेट हैcompute_metricsफंक्शन accuracy और F1 स्कोर लौटाता है- आपने पिछले अभ्यास में
training_argsपरिभाषित किए हैं
यह अभ्यास पाठ्यक्रम का हिस्सा है
PyTorch के साथ कुशल AI मॉडल प्रशिक्षण
अभ्यास निर्देश
modelकोTrainer()क्लास में पास करें.Trainer()क्लास में training arguments इनपुट करें.- मेट्रिक्स निकालने के लिए एक फंक्शन
Trainer()क्लास में पास करें. - वह डिवाइस प्रिंट करें जिसे
trainerचुनता है.
इंटरैक्टिव व्यावहारिक अभ्यास
इस अभ्यास को इस नमूना कोड को पूरा करके आज़माएँ।
trainer = Trainer(
# Pass in the model
model=____,
# Input the training arguments
args=____,
train_dataset=dataset["train"],
eval_dataset=dataset["validation"],
# Pass in a function to compute metrics
compute_metrics=____,
)
trainer.train()
# Print the device that the trainer chooses
print(trainer.args.____)