We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9bd5fd6 commit 58029f2Copy full SHA for 58029f2
train.py
@@ -66,8 +66,6 @@
66
67
# 모델 평가
68
val_loss, _, miou, _ = eval.evaluate(model, testloader, config['num_classes'], device)
69
-
70
- # Tensorboard에 평가 결과와 lr 기록
71
writer.add_scalar('Validation Loss', val_loss, epoch)
72
writer.add_scalar('mIoU', miou, epoch)
73
writer.add_text('lr', optimizer.param_groups[0]['lr'], epoch)
0 commit comments