PyTorch Lightning এবং TensorBoard ব্যবহার করে মডেল ট্রেনিংয়ের ভিজুয়ালাইজেশন করা একটি কার্যকরী উপায়, যা প্রশিক্ষণের সময় মেট্রিক্স এবং ফলাফল দেখতে সহায়তা করে। নিচে PyTorch Lightning এবং TensorBoard ব্যবহার করে মডেল ট্রেনিংয়ের একটি উদাহরণ দেওয়া হলো।
১. প্রয়োজনীয় লাইব্রেরি ইমপোর্ট করা
প্রথমে প্রয়োজনীয় লাইব্রেরিগুলি ইমপোর্ট করুন।
import pytorch_lightning as pl
import torch
from torch import nn, optim
from torch.utils.data import DataLoader, TensorDataset
from pytorch_lightning.loggers import TensorBoardLogger
২. Lightning Module তৈরি করা
Lightning Module তৈরি করুন, যেখানে মডেল, ট্রেনিং লজিক, এবং অপটিমাইজার সেটআপ করা হবে।
class LitModel(pl.LightningModule):
def __init__(self):
super(LitModel, self).__init__()
self.model = nn.Linear(10, 1) # ইনপুট সাইজ 10, আউটপুট সাইজ 1
self.criterion = nn.MSELoss() # লস ফাংশন
def forward(self, x):
return self.model(x)
def training_step(self, batch, batch_idx):
x, y = batch
y_hat = self.forward(x) # মডেল থেকে পূর্বাভাস পাওয়া
loss = self.criterion(y_hat, y) # লস হিসাব করা
self.log('train_loss', loss) # লস লগ করা
return loss
def configure_optimizers(self):
return optim.Adam(self.model.parameters(), lr=0.001) # অপটিমাইজার
৩. ডেটাসেট তৈরি করা
কাস্টম ডেটাসেট তৈরি করুন এবং DataLoader ব্যবহার করুন।
# ডেটা তৈরি করা
x_train = torch.randn(1000, 10) # 1000 টির মধ্যে 10টি বৈশিষ্ট্য
y_train = torch.randn(1000, 1) # লক্ষ্য পরিবর্তনশীল
train_dataset = TensorDataset(x_train, y_train)
train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True) # ব্যাচ সাইজ 32
৪. TensorBoard Logger তৈরি করা
TensorBoard Logger ব্যবহার করে লগিং সেটআপ করুন।
# TensorBoard Logger তৈরি করা
logger = TensorBoardLogger('logs/', name='my_model')
৫. Trainer ব্যবহার করে মডেল ট্রেনিং
Lightning Trainer ব্যবহার করে মডেলটি প্রশিক্ষণ দিন।
# Lightning মডেল তৈরি করা
model = LitModel()
# Trainer তৈরি করা
trainer = pl.Trainer(max_epochs=10, logger=logger)
# মডেল প্রশিক্ষণ দেওয়া
trainer.fit(model, train_loader)
৬. TensorBoard চালানো
TensorBoard ব্যবহার করে লগগুলি দেখতে হবে। নিচের কমান্ডটি চালান আপনার লগ ডিরেক্টরির পথ অনুযায়ী:
tensorboard --logdir=logs/
৭. ভিজ্যুয়ালাইজেশন
- আপনার ব্রাউজারে http://localhost:6006 এ যান।
- "Scalars" ট্যাবের মাধ্যমে প্রশিক্ষণের সময় লগ করা মেট্রিক্স যেমন
train_lossদেখুন।
উপসংহার
এই উদাহরণটি PyTorch Lightning এবং TensorBoard ব্যবহার করে মডেল ট্রেনিংয়ের ভিজ্যুয়ালাইজেশন কিভাবে করা যায় তা প্রদর্শন করে। TensorBoard আপনাকে প্রশিক্ষণের সময় বিভিন্ন মেট্রিক্স এবং ফলাফল পর্যবেক্ষণ করতে সহায়তা করে, যা আপনার মডেলকে উন্নত করতে সাহায্য করবে।
Read more
