ModelScope我看代码里有metric但是为啥日志只保存了loss?
可能是因为在训练过程中,metric没有被正确地计算和记录。你可以尝试以下方法来解决这个问题:
确保你的模型定义中包含了计算metric的代码。例如,如果你使用的是PyTorch,可以在forward
函数中添加计算metric的代码。
在训练循环中,确保将metric的值添加到日志中。例如,如果你使用的是torch.utils.tensorboard
,可以这样做:
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter()
for epoch in range(num_epochs):
# ...训练过程...
# 计算metric
metric_value = compute_metric(output, target)
# 将metric值添加到日志中
writer.add_scalar('Metric/YourMetricName', metric_value, epoch)
writer.close()
关闭日志文件。