Lab 11 Convolutional Neural Network :: Visdom

Uomnf97·2021년 7월 19일
0
post-thumbnail

Visdom

Visdom 설치 방법
1. pip install visdom

2. python –m visdom.server

  • visdom 사용시 주의 사항 : 사용할 동안 켜놔야 함
  • line함수에서 X값을 설정을 안해줄 경우, 0~1 사이의 임의의 값이 설정됨.

CNN Visdom에 적용

import visdom

vis = visdom.Visdom()
vis.close(env="main")

loss_plt = vis.line(Y=torch.Tensor(1).zero_(),opts=dict(title='loss_tracker', legend=['loss'], showlegend=True))

#training
total_batch = len(data_loader)

for epoch in range(training_epochs):
    avg_cost = 0
    
    for X, Y in data_loader:
        X = X.to(device)
        Y = Y.to(device)
        
        optimizxer.zero_grad()
        hypothesis = model(X)
        
        cost = criterion(hypothesis, Y)
        cost.backward()
        optimizer.step()
        
        avg_cost += cost / total_batch
    
    print('[Epoch:{}] cost = {}'.format(epoch+1, avg_cost))
    loss_tracker(loss_plt, torch.Tensor([avg_cost]), torch.Tensor([epoch]))
print('Learning Finished!')
profile
사회적 가치를 실현하는 프로그래머

0개의 댓글