[PyTorch] Livelossplot 사용예제

www.kaggle.com/pmigdal/livelossplot-for-training-loss-tracking

 

Livelossplot은 학습 과정에서 real-time으로 실행 로그를 보여주는 패키지이다.

 

다음과 같이 설치할 수 있다.

!pip install livelossplot

 

아래는 사용 예제이다.

from livelossplot import PlotLosses

liveloss = PlotLosses()
model = model.to(device)
loss_F = nn.MSELoss()
    
for epoch in range(num_epochs):
    logs = {}
    for phase in ['train', 'validation']:
        if phase == 'train': model.train()
        else: model.eval()

        running_loss = 0.0

        for x, y in dataloaders[phase]:
            x, y = x.to(device), y.to(device)

            y_pred = model(x)
            loss = loss_f(y_pred, y)

            if phase == 'train':
                optimizer.zero_grad()
                loss.backward()
                optimizer.step()

            running_loss += loss.item() * inputs.size(0)

        epoch_loss = running_loss / len(dataloaders[phase].dataset)
            
        prefix = ''
        if phase == 'validation':
            prefix = 'val_'

        logs[prefix + 'loss'] = np.sqrt(epoch_loss)
        
    liveloss.update(logs)
    liveloss.draw()