fix val dice class tensorboard name

zdaiot · Oct 12, 2019 · 1e06713 · 1e06713
1 parent 265de06
commit 1e06713
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_segment.py b/train_segment.py
@@ -133,7 +133,7 @@ def train(self, train_loader, valid_loader):
             self.writer.add_scalar('valid_loss', loss_valid, epoch)
             self.writer.add_scalar('valid_dice', dice_valid, epoch)
             for each_class, dice_class in enumerate(dice_classes):
-                self.writer.add_scalar('valid_dice_class{}'.format(each_class), dice_class, epoch)
+                self.writer.add_scalar('valid_dice_class{}'.format(each_class+1), dice_class, epoch)
 
     def validation(self, valid_loader):
         ''' 完成模型的验证过程